Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -111,11 +111,11 @@ class BasicAgent:
|
|
| 111 |
api_key=OPENROUTER_API_KEY,
|
| 112 |
)
|
| 113 |
"""
|
| 114 |
-
model_id = "ollama_chat/
|
| 115 |
model = LiteLLMModel(
|
| 116 |
model_id=model_id,
|
| 117 |
-
|
| 118 |
-
num_ctx=8192
|
| 119 |
)
|
| 120 |
tools = [
|
| 121 |
DuckDuckGoSearchTool(),
|
|
|
|
| 111 |
api_key=OPENROUTER_API_KEY,
|
| 112 |
)
|
| 113 |
"""
|
| 114 |
+
model_id = "ollama_chat/qwen2:7b"
|
| 115 |
model = LiteLLMModel(
|
| 116 |
model_id=model_id,
|
| 117 |
+
api_base="http://127.0.0.1:11434",
|
| 118 |
+
num_ctx=8192,
|
| 119 |
)
|
| 120 |
tools = [
|
| 121 |
DuckDuckGoSearchTool(),
|