diff --git a/bot/retrievers/forum_summary_retriever.py b/bot/retrievers/forum_summary_retriever.py index 8a3c16c..75df242 100644 --- 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 index cfd9cb3..c0d9b37 100644 --- 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 index 6c73bc3..92448c4 100644 --- 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 index f580c82..9b44f3c 100644 --- 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 index 0575a4b..3f027db 100644 --- 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):