Update agent.py
Browse files
agent.py
CHANGED
@@ -149,6 +149,7 @@ def build_graph(provider: str = "huggingface"):
|
|
149 |
# Groq https://console.groq.com/docs/models
|
150 |
chat = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
|
151 |
elif provider == "openai":
|
|
|
152 |
model_openai = "gpt-4o"
|
153 |
chat = ChatOpenAI(
|
154 |
model=model_openai,
|
|
|
149 |
# Groq https://console.groq.com/docs/models
|
150 |
chat = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
|
151 |
elif provider == "openai":
|
152 |
+
# Set the model from openai here
|
153 |
model_openai = "gpt-4o"
|
154 |
chat = ChatOpenAI(
|
155 |
model=model_openai,
|