Coool2 commited on
Commit
3ae963b
·
verified ·
1 Parent(s): f3fbfa1

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +4 -0
agent.py CHANGED
@@ -23,6 +23,7 @@ from llama_index.callbacks.wandb import WandbCallbackHandler
23
  from transformers import AutoModelForCausalLM, AutoTokenizer
24
  from llama_index.llms.huggingface import HuggingFaceLLM
25
  import requests
 
26
 
27
  wandb_callback = WandbCallbackHandler(run_args={"project": "gaia-llamaindex-agents"})
28
  llama_debug = LlamaDebugHandler(print_trace_on_end=True)
@@ -33,6 +34,9 @@ callback_manager = CallbackManager([
33
  llama_debug # For general debugging
34
  ])
35
 
 
 
 
36
 
37
  model_id = "Qwen/Qwen2.5-7B-Instruct"
38
  proj_llm = HuggingFaceLLM(
 
23
  from transformers import AutoModelForCausalLM, AutoTokenizer
24
  from llama_index.llms.huggingface import HuggingFaceLLM
25
  import requests
26
+ import logging
27
 
28
  wandb_callback = WandbCallbackHandler(run_args={"project": "gaia-llamaindex-agents"})
29
  llama_debug = LlamaDebugHandler(print_trace_on_end=True)
 
34
  llama_debug # For general debugging
35
  ])
36
 
37
+ logging.basicConfig(level=logging.INFO)
38
+ logging.getLogger("llama_index.core.agent").setLevel(logging.DEBUG)
39
+ logging.getLogger("llama_index.llms").setLevel(logging.DEBUG)
40
 
41
  model_id = "Qwen/Qwen2.5-7B-Instruct"
42
  proj_llm = HuggingFaceLLM(