Import fixes to get both flask and docker running

UIUC-Chatbot · Nov 28, 2023 · d726a8a · d726a8a
1 parent f98f141
commit d726a8a
Show file tree

Hide file tree

Showing 7 changed files with 13 additions and 13 deletions.
diff --git a/ai_ta_backend/agents/customcallbacks.py b/ai_ta_backend/agents/customcallbacks.py
@@ -7,7 +7,7 @@
 from langchain.callbacks import tracing_enabled
 from langchain.llms import OpenAI
 from langchain import hub
-from ai_ta_backend.agents.utils import SupabaseDB, get_langsmith_id
+from utils import SupabaseDB, get_langsmith_id
 
 
 class CustomCallbackHandler(BaseCallbackHandler):

diff --git a/ai_ta_backend/agents/github_agent.py b/ai_ta_backend/agents/github_agent.py
@@ -20,8 +20,8 @@
 # from langchain_experimental.autonomous_agents.baby_agi import BabyAGI
 from langsmith import Client
 
-from ai_ta_backend.agents.tools import get_tools
-from ai_ta_backend.agents.utils import fancier_trim_intermediate_steps
+from tools import get_tools
+from utils import fancier_trim_intermediate_steps
 
 load_dotenv(override=True, dotenv_path='.env')
 

diff --git a/ai_ta_backend/agents/github_webhook_handlers.py b/ai_ta_backend/agents/github_webhook_handlers.py
@@ -24,9 +24,9 @@
 # from langchain.tools.github.utils import generate_branch_name
 from newrelic_telemetry_sdk import Log, LogClient
 
-from ai_ta_backend.agents.github_agent import GH_Agent
-from ai_ta_backend.agents.ml4bio_agent import WorkflowAgent
-from ai_ta_backend.agents.utils import get_langsmith_trace_sharable_url
+from github_agent import GH_Agent
+from ml4bio_agent import WorkflowAgent
+from utils import get_langsmith_trace_sharable_url
 
 # load API keys from globally-available .env file
 load_dotenv(override=True, dotenv_path='.env')

diff --git a/ai_ta_backend/agents/ml4bio_agent.py b/ai_ta_backend/agents/ml4bio_agent.py
@@ -17,10 +17,10 @@
 from langchain.tools import BaseTool
 
 from langchain_experimental.plan_and_execute.executors.base import ChainExecutor
-from ai_ta_backend.agents.tools import get_tools
-from ai_ta_backend.agents.utils import fancier_trim_intermediate_steps
-from ai_ta_backend.agents.utils import SupabaseDB
-from ai_ta_backend.agents.customcallbacks import CustomCallbackHandler
+from tools import get_tools
+from utils import fancier_trim_intermediate_steps
+from utils import SupabaseDB
+from customcallbacks import CustomCallbackHandler
 
 HUMAN_MESSAGE_TEMPLATE = """Previous steps: {previous_steps}
 

diff --git a/ai_ta_backend/agents/tools.py b/ai_ta_backend/agents/tools.py
@@ -17,7 +17,7 @@
                                               create_sync_playwright_browser)
 from langchain.utilities.github import GitHubAPIWrapper
 
-from ai_ta_backend.agents.vector_db import get_vectorstore_retriever_tool
+from vector_db import get_vectorstore_retriever_tool
 from langchain.chat_models import ChatOpenAI, AzureChatOpenAI
 
 load_dotenv(override=True, dotenv_path='../.env')

diff --git a/ai_ta_backend/agents/webhooks.py b/ai_ta_backend/agents/webhooks.py
@@ -13,7 +13,7 @@
 import uuid
 from newrelic_telemetry_sdk import Log, LogClient
 
-from ai_ta_backend.agents.utils import get_supabase_client, get_langsmith_id
+from .utils import get_supabase_client, get_langsmith_id
 
 load_dotenv(override=True, dotenv_path='.env')
 

diff --git a/ai_ta_backend/main.py b/ai_ta_backend/main.py
@@ -11,7 +11,7 @@
 
 from ai_ta_backend.vector_database import Ingest
 from ai_ta_backend.web_scrape import WebScrape, mit_course_download
-from ai_ta_backend.agents import webhooks
+from .agents import webhooks
 
 import newrelic.agent
 newrelic.agent.initialize()