Coool2 commited on
Commit
370954a
·
verified ·
1 Parent(s): e6723aa

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
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