Spaces:
Configuration error
Configuration error
oremaz
commited on
Commit
·
c7e9e15
1
Parent(s):
04ea3b1
Update agent.py
Browse files
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(
|