Update ai_agent.py

AreaLayer · Feb 12, 2024 · 55850cf · 55850cf
1 parent c092ab4
commit 55850cf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/core/ai_agent.py b/src/core/ai_agent.py
@@ -3,6 +3,7 @@
 
 from llama_index.llms import OpenAI, ChatMessage
 from llama_index.tools import BaseTool, FunctionTool
+from lightning import Data, Channel
 
 import nest_asyncio