Spaces:
Configuration error
Configuration error
oremaz
commited on
Commit
·
1dd42da
1
Parent(s):
4755ec7
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -21,6 +21,7 @@ import wandb
|
|
| 21 |
from llama_index.callbacks.wandb import WandbCallbackHandler
|
| 22 |
from llama_index.core.callbacks.base import CallbackManager
|
| 23 |
from llama_index.core.callbacks.llama_debug import LlamaDebugHandler
|
|
|
|
| 24 |
|
| 25 |
proj_llm = OpenRouter(
|
| 26 |
model="mistralai/mistral-small-3.1-24b-instruct:free",
|
|
@@ -34,7 +35,6 @@ wandb_callback = WandbCallbackHandler(run_args={"project": "gaia-llamaindex-agen
|
|
| 34 |
llama_debug = LlamaDebugHandler(print_trace_on_end=True)
|
| 35 |
callback_manager = CallbackManager([wandb_callback, llama_debug])
|
| 36 |
|
| 37 |
-
|
| 38 |
Settings.llm = proj_llm
|
| 39 |
Settings.embed_model = embed_model
|
| 40 |
Settings.callback_manager = callback_manager
|
|
|
|
| 21 |
from llama_index.callbacks.wandb import WandbCallbackHandler
|
| 22 |
from llama_index.core.callbacks.base import CallbackManager
|
| 23 |
from llama_index.core.callbacks.llama_debug import LlamaDebugHandler
|
| 24 |
+
from llama_index.core import Settings
|
| 25 |
|
| 26 |
proj_llm = OpenRouter(
|
| 27 |
model="mistralai/mistral-small-3.1-24b-instruct:free",
|
|
|
|
| 35 |
llama_debug = LlamaDebugHandler(print_trace_on_end=True)
|
| 36 |
callback_manager = CallbackManager([wandb_callback, llama_debug])
|
| 37 |
|
|
|
|
| 38 |
Settings.llm = proj_llm
|
| 39 |
Settings.embed_model = embed_model
|
| 40 |
Settings.callback_manager = callback_manager
|