Spaces:
Runtime error
Runtime error
Upload agent.py
Browse files
agent.py
CHANGED
|
@@ -177,11 +177,11 @@ tools = [
|
|
| 177 |
]
|
| 178 |
|
| 179 |
# Build graph function
|
| 180 |
-
def build_graph(provider: str = "
|
| 181 |
"""Build the graph"""
|
| 182 |
# Load environment variables from .env file
|
| 183 |
if provider == "openai":
|
| 184 |
-
llm = ChatOpenAI(model_name="gpt-
|
| 185 |
elif provider == "anthropic":
|
| 186 |
llm = ChatAnthropic(model="claude-v1", temperature=0)
|
| 187 |
elif provider == "google":
|
|
|
|
| 177 |
]
|
| 178 |
|
| 179 |
# Build graph function
|
| 180 |
+
def build_graph(provider: str = "openai"):
|
| 181 |
"""Build the graph"""
|
| 182 |
# Load environment variables from .env file
|
| 183 |
if provider == "openai":
|
| 184 |
+
llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0)
|
| 185 |
elif provider == "anthropic":
|
| 186 |
llm = ChatAnthropic(model="claude-v1", temperature=0)
|
| 187 |
elif provider == "google":
|