Upload 6 files

agents/graph.py

@@ -1,6 +1,6 @@
 import os
 import logging
-from tools_and_schemas import SearchQueryList, Reflection
+from agents.tools_and_schemas import SearchQueryList, Reflection
 from dotenv import load_dotenv
 from langchain_core.messages import AIMessage
 from langgraph.types import Send
@@ -9,14 +9,14 @@ from langgraph.graph import START, END
 from langchain_core.runnables import RunnableConfig
 from google.genai import Client
 
-from state import (
+from agents.state import (
     OverallState,
     QueryGenerationState,
     ReflectionState,
     WebSearchState,
 )
-from configuration import Configuration
-from prompts import (
+from agents.configuration import Configuration
+from agents.prompts import (
     get_current_date,
     query_writer_instructions,
     web_searcher_instructions,
@@ -25,7 +25,7 @@ from prompts import (
     gaia_system_instructions,
 )
 from langchain_google_genai import ChatGoogleGenerativeAI
-from utils import (
+from agents.utils import (
     get_citations,
     get_research_topic,
     insert_citation_markers,