Spaces:
Sleeping
Sleeping
Update LangGraphAgent to use ChatTogether model instead of ChatOpenRouter
Browse files
agent.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
import os
|
| 2 |
from typing import Optional
|
| 3 |
-
|
| 4 |
from langgraph.graph import StateGraph, START, END, MessagesState
|
| 5 |
from langchain_core.messages import HumanMessage, AIMessage, SystemMessage
|
| 6 |
from langchain_google_genai import ChatGoogleGenerativeAI
|
|
@@ -77,8 +77,10 @@ class LangGraphAgent:
|
|
| 77 |
# model = ChatGoogleGenerativeAI(
|
| 78 |
# model="gemma-3-27b-it",
|
| 79 |
# )
|
| 80 |
-
model =
|
| 81 |
-
model="
|
|
|
|
|
|
|
| 82 |
)
|
| 83 |
if model is None and ChatOpenAI is not None:
|
| 84 |
model = ChatOpenAI(
|
|
|
|
| 1 |
import os
|
| 2 |
from typing import Optional
|
| 3 |
+
from langchain_together import ChatTogether
|
| 4 |
from langgraph.graph import StateGraph, START, END, MessagesState
|
| 5 |
from langchain_core.messages import HumanMessage, AIMessage, SystemMessage
|
| 6 |
from langchain_google_genai import ChatGoogleGenerativeAI
|
|
|
|
| 77 |
# model = ChatGoogleGenerativeAI(
|
| 78 |
# model="gemma-3-27b-it",
|
| 79 |
# )
|
| 80 |
+
model = ChatTogether(
|
| 81 |
+
model="openai/gpt-oss-20b",
|
| 82 |
+
api_key=os.getenv("TOGETHER_API_KEY"),
|
| 83 |
+
|
| 84 |
)
|
| 85 |
if model is None and ChatOpenAI is not None:
|
| 86 |
model = ChatOpenAI(
|