Spaces:
Sleeping
Sleeping
Jialun He
commited on
Commit
·
c97e291
1
Parent(s):
1ff2bf8
Update agent.py
Browse files
agent.py
CHANGED
@@ -159,7 +159,7 @@ tools = [
|
|
159 |
]
|
160 |
|
161 |
# Build graph function
|
162 |
-
def build_graph(provider: str = "
|
163 |
"""Build the graph"""
|
164 |
# Load environment variables from .env file
|
165 |
if provider == "google":
|
@@ -170,7 +170,7 @@ def build_graph(provider: str = "google"):
|
|
170 |
llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
|
171 |
elif provider == "openai":
|
172 |
# OpenAI
|
173 |
-
llm = ChatOpenAI(model="gpt-
|
174 |
elif provider == "huggingface":
|
175 |
llm = ChatHuggingFace(
|
176 |
llm=HuggingFaceEndpoint(
|
|
|
159 |
]
|
160 |
|
161 |
# Build graph function
|
162 |
+
def build_graph(provider: str = "openai"):
|
163 |
"""Build the graph"""
|
164 |
# Load environment variables from .env file
|
165 |
if provider == "google":
|
|
|
170 |
llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
|
171 |
elif provider == "openai":
|
172 |
# OpenAI
|
173 |
+
llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)
|
174 |
elif provider == "huggingface":
|
175 |
llm = ChatHuggingFace(
|
176 |
llm=HuggingFaceEndpoint(
|