trongld commited on
Commit
713244e
·
1 Parent(s): a37c30b

Update LangGraphAgent to use ChatTogether model instead of ChatOpenRouter

Browse files
Files changed (1) hide show
  1. agent.py +5 -3
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 = ChatOpenRouter(
81
- model="z-ai/glm-4.5-air:free",
 
 
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(