Coool2 commited on
Commit
bac751d
·
verified ·
1 Parent(s): ead68d3

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -1
agent.py CHANGED
@@ -18,12 +18,13 @@ from llama_index.callbacks.wandb import WandbCallbackHandler
18
  from llama_index.core.callbacks.base import CallbackManager
19
  from llama_index.core.callbacks.llama_debug import LlamaDebugHandler
20
  from llama_index.core import Settings
21
- from llama_index.callbacks.wandb import WandbCallbackHandler
22
 
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)
 
18
  from llama_index.core.callbacks.base import CallbackManager
19
  from llama_index.core.callbacks.llama_debug import LlamaDebugHandler
20
  from llama_index.core import Settings
 
21
 
22
  from transformers import AutoModelForCausalLM, AutoTokenizer
23
  from llama_index.llms.huggingface import HuggingFaceLLM
24
  import requests
25
  import logging
26
+ from llama_index.core.workflow import Context
27
+ from llama_index.core.agent.workflow import AgentStream
28
 
29
  wandb_callback = WandbCallbackHandler(run_args={"project": "gaia-llamaindex-agents"})
30
  llama_debug = LlamaDebugHandler(print_trace_on_end=True)