trongld commited on
Commit
075ffbc
·
1 Parent(s): ba54b7f

Update LangGraphAgent to use ChatOpenRouter model instead of ChatGoogleGenerativeAI

Browse files
Files changed (1) hide show
  1. agent.py +6 -2
agent.py CHANGED
@@ -74,8 +74,12 @@ class LangGraphAgent:
74
 
75
  # Choose an LLM if not provided
76
  if model is None:
77
- model = ChatGoogleGenerativeAI(
78
- model="gemma-3-27b-it",
 
 
 
 
79
  )
80
  if model is None and ChatOpenAI is not None:
81
  model = ChatOpenAI(
 
74
 
75
  # Choose an LLM if not provided
76
  if model is None:
77
+ # model = ChatGoogleGenerativeAI(
78
+ # model="gemma-3-27b-it",
79
+ # )
80
+ model = ChatOpenRouter(
81
+ model="openai/gpt-oss-20b:free",
82
+ temperature=0.1,
83
  )
84
  if model is None and ChatOpenAI is not None:
85
  model = ChatOpenAI(