38 lines
929 B
Python
38 lines
929 B
Python
|
import os
|
||
|
|
from langchain_anthropic import ChatAnthropic
|
||
|
|
from langchain_openai import ChatOpenAI
|
||
|
|
from langchain_google_genai import ChatGoogleGenerativeAI
|
||
|
|
|
||
|
|
|
||
|
|
def get_orchestrator():
|
||
|
|
return ChatAnthropic(model="claude-sonnet-4-20250514", temperature=0)
|
||
|
|
|
||
|
|
|
||
|
|
def get_implementer():
|
||
|
|
return ChatAnthropic(model="claude-sonnet-4-20250514", temperature=0)
|
||
|
|
|
||
|
|
|
||
|
|
def get_reviewer():
|
||
|
|
return ChatAnthropic(model="claude-sonnet-4-20250514", temperature=0)
|
||
|
|
|
||
|
|
|
||
|
|
def get_researcher():
|
||
|
|
return ChatAnthropic(model="claude-haiku-4-5-20251001", temperature=0)
|
||
|
|
|
||
|
|
|
||
|
|
def get_cross_reviewer():
|
||
|
|
return ChatOpenAI(model="gpt-4.1", temperature=0.2)
|
||
|
|
|
||
|
|
|
||
|
|
def get_scanner():
|
||
|
|
return ChatGoogleGenerativeAI(model="gemini-2.5-pro", temperature=0)
|
||
|
|
|
||
|
|
|
||
|
|
def get_fast_coder():
|
||
|
|
return ChatOpenAI(
|
||
|
|
model="deepseek-coder",
|
||
|
|
base_url="https://api.deepseek.com/v1",
|
||
|
|
api_key=os.getenv("DEEPSEEK_API_KEY"),
|
||
|
|
temperature=0,
|
||
|
|
)
|