feat: change imports

agent/graph.py

@@ -1,6 +1,6 @@
-from custom_state import AssistantState
+from agent.custom_state import AssistantState
 from langgraph.graph import START, StateGraph, END
-from nodes import assisstant, tools, get_file
+from agent.nodes import assisstant, tools, get_file
 from langgraph.prebuilt import tools_condition, ToolNode
 
 builder = StateGraph(AssistantState)

agent/nodes.py

@@ -5,13 +5,13 @@ from youtube_transcript_api import YouTubeTranscriptApi
 from youtube_transcript_api._errors import TranscriptsDisabled
 from langchain_core.messages import SystemMessage, HumanMessage
 from tavily import TavilyClient
-from custom_state import AssistantState
+from agent.custom_state import AssistantState
 import os
-from models import openai_model, gemini_model, llama_client, openai_audio_model
+from agent.models import openai_model, gemini_model, llama_client, openai_audio_model
 from dotenv import load_dotenv
 from e2b_code_interpreter import Sandbox
-from utils import load_data, extract_python
-from prompts import prompt_coding
+from agent.utils import load_data, extract_python
+from agent.prompts import prompt_coding
 
 import requests
 

agent/utils.py

@@ -1,7 +1,7 @@
 from langchain_community.document_loaders.csv_loader import CSVLoader
 from langchain_community.document_loaders import PyPDFLoader, UnstructuredExcelLoader
 from langchain_community.document_loaders.python import PythonLoader
-from data import FileLoader
+from agent.data import FileLoader
 import base64
 import re, textwrap