feat: update imports based on code restructuring!

TogetherCrew · Dec 28, 2023 · f4c9980 · f4c9980
1 parent 0d6e038
commit f4c9980
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 7 deletions.
diff --git a/bot/retrievers/forum_summary_retriever.py b/bot/retrievers/forum_summary_retriever.py
@@ -1,5 +1,5 @@
 from llama_index.embeddings import BaseEmbedding
-from retrievers.summary_retriever_base import BaseSummarySearch
+from bot.retrievers.summary_retriever_base import BaseSummarySearch
 from tc_hivemind_backend.embeddings.cohere import CohereEmbedding
 
 

diff --git a/discord_query.py b/discord_query.py
@@ -1,8 +1,8 @@
 from llama_index import QueryBundle
 from llama_index.vector_stores import ExactMatchFilter, FilterCondition, MetadataFilters
-from retrievers.forum_summary_retriever import ForumBasedSummaryRetriever
-from retrievers.process_dates import process_dates
-from retrievers.utils.load_hyperparams import load_hyperparams
+from bot.retrievers.forum_summary_retriever import ForumBasedSummaryRetriever
+from bot.retrievers.process_dates import process_dates
+from bot.retrievers.utils.load_hyperparams import load_hyperparams
 from tc_hivemind_backend.embeddings.cohere import CohereEmbedding
 from tc_hivemind_backend.pg_vector_access import PGVectorAccess
 

diff --git a/tests/unit/test_discord_summary_retriever.py b/tests/unit/test_discord_summary_retriever.py
@@ -5,7 +5,7 @@
 
 from dateutil import parser
 from llama_index import Document, MockEmbedding, ServiceContext, VectorStoreIndex
-from retrievers.forum_summary_retriever import ForumBasedSummaryRetriever
+from bot.retrievers.forum_summary_retriever import ForumBasedSummaryRetriever
 
 
 class TestDiscordSummaryRetriever(TestCase):

diff --git a/tests/unit/test_process_dates_forum_retriever_search.py b/tests/unit/test_process_dates_forum_retriever_search.py
@@ -1,6 +1,6 @@
 import unittest
 
-from retrievers.process_dates import process_dates
+from bot.retrievers.process_dates import process_dates
 
 
 class TestProcessDates(unittest.TestCase):

diff --git a/tests/unit/test_summary_retriever_base.py b/tests/unit/test_summary_retriever_base.py
@@ -4,7 +4,7 @@
 
 from llama_index import Document, MockEmbedding, ServiceContext, VectorStoreIndex
 from llama_index.schema import NodeWithScore
-from retrievers.summary_retriever_base import BaseSummarySearch
+from bot.retrievers.summary_retriever_base import BaseSummarySearch
 
 
 class TestSummaryRetrieverBase(TestCase):