deepsense-ai · mhordynski · Sep 17, 2024 · Sep 16, 2024 · Sep 16, 2024
diff --git a/packages/ragbits-common/examples/llm_example.py b/packages/ragbits-common/examples/llm_example.py
@@ -2,8 +2,8 @@
 
 from pydantic import BaseModel
 
-from ragbits_common.llms.litellm import LiteLLM
-from ragbits_common.prompt import Prompt
+from ragbits.common.llms.litellm import LiteLLM
+from ragbits.common.prompt import Prompt
 
 
 class LoremPromptInput(BaseModel):

diff --git a/packages/ragbits-common/examples/prompt_example.py b/packages/ragbits-common/examples/prompt_example.py
@@ -1,6 +1,6 @@
 from pydantic import BaseModel
 
-from ragbits_common.prompt import Prompt
+from ragbits.common.prompt import Prompt
 
 
 class LoremPromptInput(BaseModel):

diff --git a/packages/ragbits-common/setup.cfg b/packages/ragbits-common/setup.cfg
@@ -22,7 +22,7 @@ classifiers =
 [options]
 package_dir=
     =src
-packages=find:
+packages=find_namespace:
 zip_safe = False
 platforms = any
 include_package_data = True

diff --git a/...its-common/src/ragbits_common/__init__.py → ...its-common/src/ragbits/common/__init__.py b/...its-common/src/ragbits_common/__init__.py → ...its-common/src/ragbits/common/__init__.py
diff --git a/...src/ragbits_common/embeddings/__init__.py → ...src/ragbits/common/embeddings/__init__.py b/...src/ragbits_common/embeddings/__init__.py → ...src/ragbits/common/embeddings/__init__.py
diff --git a/...mon/src/ragbits_common/embeddings/base.py → ...mon/src/ragbits/common/embeddings/base.py b/...mon/src/ragbits_common/embeddings/base.py → ...mon/src/ragbits/common/embeddings/base.py
diff --git a/...c/ragbits_common/embeddings/exceptions.py → ...c/ragbits/common/embeddings/exceptions.py b/...c/ragbits_common/embeddings/exceptions.py → ...c/ragbits/common/embeddings/exceptions.py
diff --git a/.../src/ragbits_common/embeddings/litellm.py → .../src/ragbits/common/embeddings/litellm.py b/.../src/ragbits_common/embeddings/litellm.py → .../src/ragbits/common/embeddings/litellm.py
@@ -7,8 +7,8 @@
 except ImportError:
     HAS_LITELLM = False
 
-from ragbits_common.embeddings.base import Embeddings
-from ragbits_common.embeddings.exceptions import EmbeddingConnectionError, EmbeddingResponseError, EmbeddingStatusError
+from ragbits.common.embeddings.base import Embeddings
+from ragbits.common.embeddings.exceptions import EmbeddingConnectionError, EmbeddingResponseError, EmbeddingStatusError
 
 
 class LiteLLMEmbeddings(Embeddings):

diff --git a/...ommon/src/ragbits_common/llms/__init__.py → ...ommon/src/ragbits/common/llms/__init__.py b/...ommon/src/ragbits_common/llms/__init__.py → ...ommon/src/ragbits/common/llms/__init__.py
diff --git a/...ts-common/src/ragbits_common/llms/base.py → ...ts-common/src/ragbits/common/llms/base.py b/...ts-common/src/ragbits_common/llms/base.py → ...ts-common/src/ragbits/common/llms/base.py
@@ -2,7 +2,7 @@
 from functools import cached_property
 from typing import Generic, Optional, Type, cast, overload
 
-from ragbits_common.prompt.base import BasePrompt, BasePromptWithParser, OutputT
+from ragbits.common.prompt.base import BasePrompt, BasePromptWithParser, OutputT
 
 from .clients.base import LLMClient, LLMClientOptions, LLMOptions
 

diff --git a/...c/ragbits_common/llms/clients/__init__.py → ...c/ragbits/common/llms/clients/__init__.py b/...c/ragbits_common/llms/clients/__init__.py → ...c/ragbits/common/llms/clients/__init__.py
diff --git a/...n/src/ragbits_common/llms/clients/base.py → ...n/src/ragbits/common/llms/clients/base.py b/...n/src/ragbits_common/llms/clients/base.py → ...n/src/ragbits/common/llms/clients/base.py
@@ -4,7 +4,7 @@
 
 from pydantic import BaseModel
 
-from ragbits_common.prompt import ChatFormat
+from ragbits.common.prompt import ChatFormat
 
 from ..types import NotGiven
 

diff --git a/...ragbits_common/llms/clients/exceptions.py → ...ragbits/common/llms/clients/exceptions.py b/...ragbits_common/llms/clients/exceptions.py → ...ragbits/common/llms/clients/exceptions.py
diff --git a/...rc/ragbits_common/llms/clients/litellm.py → ...rc/ragbits/common/llms/clients/litellm.py b/...rc/ragbits_common/llms/clients/litellm.py → ...rc/ragbits/common/llms/clients/litellm.py
@@ -11,7 +11,7 @@
     HAS_LITELLM = False
 
 
-from ragbits_common.prompt import ChatFormat
+from ragbits.common.prompt import ChatFormat
 
 from ..types import NOT_GIVEN, NotGiven
 from .base import LLMClient, LLMOptions

diff --git a/.../src/ragbits_common/llms/clients/local.py → .../src/ragbits/common/llms/clients/local.py b/.../src/ragbits_common/llms/clients/local.py → .../src/ragbits/common/llms/clients/local.py
@@ -11,7 +11,7 @@
 except ImportError:
     HAS_LOCAL_LLM = False
 
-from ragbits_common.prompt import ChatFormat
+from ragbits.common.prompt import ChatFormat
 
 from ..types import NOT_GIVEN, NotGiven
 from .base import LLMClient, LLMOptions

diff --git a/...common/src/ragbits_common/llms/litellm.py → ...common/src/ragbits/common/llms/litellm.py b/...common/src/ragbits_common/llms/litellm.py → ...common/src/ragbits/common/llms/litellm.py
@@ -8,7 +8,7 @@
 except ImportError:
     HAS_LITELLM = False
 
-from ragbits_common.prompt.base import BasePrompt
+from ragbits.common.prompt.base import BasePrompt
 
 from .base import LLM
 from .clients.litellm import LiteLLMClient, LiteLLMOptions

diff --git a/...s-common/src/ragbits_common/llms/local.py → ...s-common/src/ragbits/common/llms/local.py b/...s-common/src/ragbits_common/llms/local.py → ...s-common/src/ragbits/common/llms/local.py
@@ -8,7 +8,7 @@
 except ImportError:
     HAS_LOCAL_LLM = False
 
-from ragbits_common.prompt.base import BasePrompt
+from ragbits.common.prompt.base import BasePrompt
 
 from .base import LLM
 from .clients.local import LocalLLMClient, LocalLLMOptions

diff --git a/...s-common/src/ragbits_common/llms/types.py → ...s-common/src/ragbits/common/llms/types.py b/...s-common/src/ragbits_common/llms/types.py → ...s-common/src/ragbits/common/llms/types.py
diff --git a/packages/ragbits-common/src/ragbits/common/prompt/__init__.py b/packages/ragbits-common/src/ragbits/common/prompt/__init__.py
@@ -0,0 +1,3 @@
+from ragbits.common.prompt.prompt import ChatFormat, Prompt
+
+__all__ = ["Prompt", "ChatFormat"]
diff --git a/...-common/src/ragbits_common/prompt/base.py → ...-common/src/ragbits/common/prompt/base.py b/...-common/src/ragbits_common/prompt/base.py → ...-common/src/ragbits/common/prompt/base.py
diff --git a/...mmon/src/ragbits_common/prompt/parsers.py → ...mmon/src/ragbits/common/prompt/parsers.py b/...mmon/src/ragbits_common/prompt/parsers.py → ...mmon/src/ragbits/common/prompt/parsers.py
diff --git a/...ommon/src/ragbits_common/prompt/prompt.py → ...ommon/src/ragbits/common/prompt/prompt.py b/...ommon/src/ragbits_common/prompt/prompt.py → ...ommon/src/ragbits/common/prompt/prompt.py
diff --git a/...agbits-common/src/ragbits_common/py.typed → packages/ragbits-common/src/ragbits/py.typed b/...agbits-common/src/ragbits_common/py.typed → packages/ragbits-common/src/ragbits/py.typed
diff --git a/packages/ragbits-common/src/ragbits_common/prompt/__init__.py b/packages/ragbits-common/src/ragbits_common/prompt/__init__.py
diff --git a/packages/ragbits-common/tests/unit/llms/test_litellm.py b/packages/ragbits-common/tests/unit/llms/test_litellm.py
@@ -1,9 +1,9 @@
 from pydantic import BaseModel
 
-from ragbits_common.llms.clients.litellm import LiteLLMOptions
-from ragbits_common.llms.litellm import LiteLLM
-from ragbits_common.prompt import Prompt
-from ragbits_common.prompt.base import BasePrompt, BasePromptWithParser, ChatFormat
+from ragbits.common.llms.clients.litellm import LiteLLMOptions
+from ragbits.common.llms.litellm import LiteLLM
+from ragbits.common.prompt import Prompt
+from ragbits.common.prompt.base import BasePrompt, BasePromptWithParser, ChatFormat
 
 
 class MockPrompt(BasePrompt):

diff --git a/packages/ragbits-common/tests/unit/prompts/test_parsers.py b/packages/ragbits-common/tests/unit/prompts/test_parsers.py
@@ -2,8 +2,8 @@
 
 import pytest
 
-from ragbits_common.prompt import Prompt
-from ragbits_common.prompt.parsers import ResponseParsingError
+from ragbits.common.prompt import Prompt
+from ragbits.common.prompt.parsers import ResponseParsingError
 
 from .test_prompt import _PromptOutput
 

diff --git a/packages/ragbits-common/tests/unit/prompts/test_prompt.py b/packages/ragbits-common/tests/unit/prompts/test_prompt.py
@@ -1,7 +1,7 @@
 import pydantic
 import pytest
 
-from ragbits_common.prompt import Prompt
+from ragbits.common.prompt import Prompt
 
 
 class _PromptInput(pydantic.BaseModel):

diff --git a/packages/ragbits-dev-kit/setup.cfg b/packages/ragbits-dev-kit/setup.cfg
@@ -22,7 +22,7 @@ classifiers =
 [options]
 package_dir=
     =src
-packages=find:
+packages=find_namespace:
 zip_safe = False
 platforms = any
 include_package_data = True

diff --git a/...s-dev-kit/src/ragbits_dev_kit/__init__.py → ...s-dev-kit/src/ragbits/dev_kit/__init__.py b/...s-dev-kit/src/ragbits_dev_kit/__init__.py → ...s-dev-kit/src/ragbits/dev_kit/__init__.py
diff --git a/...v-kit/src/ragbits_dev_kit/app/__init__.py → ...v-kit/src/ragbits/dev_kit/app/__init__.py b/...v-kit/src/ragbits_dev_kit/app/__init__.py → ...v-kit/src/ragbits/dev_kit/app/__init__.py
diff --git a/...s-dev-kit/src/ragbits_dev_kit/app/main.py → ...s-dev-kit/src/ragbits/dev_kit/app/main.py b/...s-dev-kit/src/ragbits_dev_kit/app/main.py → ...s-dev-kit/src/ragbits/dev_kit/app/main.py
@@ -5,10 +5,10 @@
 import jinja2
 import typer
 from pydantic import BaseModel
-from ragbits_dev_kit.discovery.prompt_discovery import PromptDiscovery
 
-from ragbits_common.llms import LiteLLM
-from ragbits_common.llms.clients import LiteLLMOptions
+from ragbits.common.llms import LiteLLM
+from ragbits.common.llms.clients import LiteLLMOptions
+from ragbits.dev_kit.discovery.prompt_discovery import PromptDiscovery
 
 
 class PromptState:

diff --git a/packages/ragbits-dev-kit/src/ragbits/dev_kit/discovery/__init__.py b/packages/ragbits-dev-kit/src/ragbits/dev_kit/discovery/__init__.py
@@ -0,0 +1,3 @@
+from ragbits.dev_kit.discovery.prompt_discovery import PromptDiscovery
+
+__all__ = ["PromptDiscovery"]
diff --git a/...its_dev_kit/discovery/prompt_discovery.py → ...its/dev_kit/discovery/prompt_discovery.py b/...its_dev_kit/discovery/prompt_discovery.py → ...its/dev_kit/discovery/prompt_discovery.py
@@ -4,7 +4,7 @@
 from collections import namedtuple
 from typing import Any
 
-from ragbits_common.prompt import Prompt
+from ragbits.common.prompt import Prompt
 
 PromptDetails = namedtuple("PromptDetails", ["system_prompt", "user_prompt", "input_type", "object"])
 

diff --git a/...arch/src/ragbits_document_search/py.typed → ...ages/ragbits-dev-kit/src/ragbits/py.typed b/...arch/src/ragbits_document_search/py.typed → ...ages/ragbits-dev-kit/src/ragbits/py.typed
diff --git a/packages/ragbits-dev-kit/src/ragbits_dev_kit/discovery/__init__.py b/packages/ragbits-dev-kit/src/ragbits_dev_kit/discovery/__init__.py
diff --git a/packages/ragbits-dev-kit/tests/unit/discovery/prompt_classes_for_tests.py b/packages/ragbits-dev-kit/tests/unit/discovery/prompt_classes_for_tests.py
@@ -2,7 +2,7 @@
 
 from pydantic import BaseModel
 
-from ragbits_common.prompt import Prompt
+from ragbits.common.prompt import Prompt
 
 
 class PromptForTestInput(BaseModel):

diff --git a/...gbits-dev-kit/tests/unit/discovery/ragbits_tests_pkg_with_prompts/prompts/temp_prompt1.py b/...gbits-dev-kit/tests/unit/discovery/ragbits_tests_pkg_with_prompts/prompts/temp_prompt1.py
@@ -1,6 +1,6 @@
 from pydantic import BaseModel
 
-from ragbits_common.prompt import Prompt
+from ragbits.common.prompt import Prompt
 
 
 class PromptForTestInputA(BaseModel):

diff --git a/...gbits-dev-kit/tests/unit/discovery/ragbits_tests_pkg_with_prompts/prompts/temp_prompt2.py b/...gbits-dev-kit/tests/unit/discovery/ragbits_tests_pkg_with_prompts/prompts/temp_prompt2.py
@@ -1,6 +1,6 @@
 from pydantic import BaseModel
 
-from ragbits_common.prompt import Prompt
+from ragbits.common.prompt import Prompt
 
 
 class PromptForTestInputB(BaseModel):

diff --git a/packages/ragbits-dev-kit/tests/unit/discovery/test_prompt_discovery.py b/packages/ragbits-dev-kit/tests/unit/discovery/test_prompt_discovery.py
@@ -1,7 +1,7 @@
 import sys
 from pathlib import Path
 
-from ragbits_dev_kit.discovery.prompt_discovery import PromptDiscovery
+from ragbits.dev_kit.discovery.prompt_discovery import PromptDiscovery
 
 
 def test_prompt_discovery_from_file():

diff --git a/packages/ragbits-document-search/examples/simple_text.py b/packages/ragbits-document-search/examples/simple_text.py
@@ -1,10 +1,9 @@
 import asyncio
 
-from ragbits_document_search import DocumentSearch
-from ragbits_document_search.documents.document import DocumentMeta
-from ragbits_document_search.vector_store.in_memory import InMemoryVectorStore
-
-from ragbits_common.embeddings.litellm import LiteLLMEmbeddings
+from ragbits.common.embeddings.litellm import LiteLLMEmbeddings
+from ragbits.document_search import DocumentSearch
+from ragbits.document_search.documents.document import DocumentMeta
+from ragbits.document_search.vector_store.in_memory import InMemoryVectorStore
 
 documents = [
     DocumentMeta.create_text_document_from_literal("RIP boiled water. You will be mist."),

diff --git a/packages/ragbits-document-search/setup.cfg b/packages/ragbits-document-search/setup.cfg
@@ -23,7 +23,7 @@ classifiers =
 [options]
 package_dir=
     =src
-packages=find:
+packages=find_namespace:
 zip_safe = False
 platforms = any
 include_package_data = True

diff --git a/...h/src/ragbits_document_search/__init__.py → ...h/src/ragbits/document_search/__init__.py b/...h/src/ragbits_document_search/__init__.py → ...h/src/ragbits/document_search/__init__.py
diff --git a/...rc/ragbits_document_search/__version__.py → ...rc/ragbits/document_search/__version__.py b/...rc/ragbits_document_search/__version__.py → ...rc/ragbits/document_search/__version__.py
diff --git a/...arch/src/ragbits_document_search/_main.py → ...arch/src/ragbits/document_search/_main.py b/...arch/src/ragbits_document_search/_main.py → ...arch/src/ragbits/document_search/_main.py
@@ -1,13 +1,12 @@
-from ragbits_document_search.documents.document import DocumentMeta
-from ragbits_document_search.documents.element import Element
-from ragbits_document_search.ingestion.document_processor import DocumentProcessor
-from ragbits_document_search.retrieval.rephrasers.base import QueryRephraser
-from ragbits_document_search.retrieval.rephrasers.noop import NoopQueryRephraser
-from ragbits_document_search.retrieval.rerankers.base import Reranker
-from ragbits_document_search.retrieval.rerankers.noop import NoopReranker
-from ragbits_document_search.vector_store.base import VectorStore
-
-from ragbits_common.embeddings.base import Embeddings
+from ragbits.common.embeddings.base import Embeddings
+from ragbits.document_search.documents.document import DocumentMeta
+from ragbits.document_search.documents.element import Element
+from ragbits.document_search.ingestion.document_processor import DocumentProcessor
+from ragbits.document_search.retrieval.rephrasers.base import QueryRephraser
+from ragbits.document_search.retrieval.rephrasers.noop import NoopQueryRephraser
+from ragbits.document_search.retrieval.rerankers.base import Reranker
+from ragbits.document_search.retrieval.rerankers.noop import NoopReranker
+from ragbits.document_search.vector_store.base import VectorStore
 
 
 class DocumentSearch:

diff --git a/...its_document_search/documents/__init__.py → ...its/document_search/documents/__init__.py b/...its_document_search/documents/__init__.py → ...its/document_search/documents/__init__.py
diff --git a/...its_document_search/documents/document.py → ...its/document_search/documents/document.py b/...its_document_search/documents/document.py → ...its/document_search/documents/document.py
@@ -4,7 +4,8 @@
 from typing import Union
 
 from pydantic import BaseModel, Field
-from ragbits_document_search.documents.sources import LocalFileSource
+
+from ragbits.document_search.documents.sources import LocalFileSource
 
 
 class DocumentType(str, Enum):

diff --git a/...bits_document_search/documents/element.py → ...bits/document_search/documents/element.py b/...bits_document_search/documents/element.py → ...bits/document_search/documents/element.py
@@ -2,8 +2,9 @@
 from typing import ClassVar
 
 from pydantic import BaseModel
-from ragbits_document_search.documents.document import DocumentMeta
-from ragbits_document_search.vector_store.base import VectorDBEntry
+
+from ragbits.document_search.documents.document import DocumentMeta
+from ragbits.document_search.vector_store.base import VectorDBEntry
 
 
 class Element(BaseModel, ABC):
@@ -26,7 +27,7 @@ def get_key(self) -> str:
         """
 
     @classmethod
-    def __pydantic_init_subclass__(cls, **kwargs):  # pylint: disable=unused-argument
+    def __pydantic_init_subclass__(cls, **kwargs: dict) -> None:  # pylint: disable=unused-argument
         element_type_default = cls.model_fields["element_type"].default
 
         if element_type_default is None:

diff --git a/...bits_document_search/documents/sources.py → ...bits/document_search/documents/sources.py b/...bits_document_search/documents/sources.py → ...bits/document_search/documents/sources.py
diff --git a/...its_document_search/ingestion/__init__.py → ...its/document_search/ingestion/__init__.py b/...its_document_search/ingestion/__init__.py → ...its/document_search/ingestion/__init__.py
diff --git a/...nt_search/ingestion/document_processor.py → ...nt_search/ingestion/document_processor.py b/...nt_search/ingestion/document_processor.py → ...nt_search/ingestion/document_processor.py
@@ -4,8 +4,8 @@
 
 from typing import List
 
-from ragbits_document_search.documents.document import DocumentMeta, TextDocument
-from ragbits_document_search.documents.element import Element, TextElement
+from ragbits.document_search.documents.document import DocumentMeta, TextDocument
+from ragbits.document_search.documents.element import Element, TextElement
 
 
 class DocumentProcessor:

diff --git a/...its_document_search/retrieval/__init__.py → ...its/document_search/retrieval/__init__.py b/...its_document_search/retrieval/__init__.py → ...its/document_search/retrieval/__init__.py
diff --git a/...t_search/retrieval/rephrasers/__init__.py → ...t_search/retrieval/rephrasers/__init__.py b/...t_search/retrieval/rephrasers/__init__.py → ...t_search/retrieval/rephrasers/__init__.py
diff --git a/...ument_search/retrieval/rephrasers/base.py → ...ument_search/retrieval/rephrasers/base.py b/...ument_search/retrieval/rephrasers/base.py → ...ument_search/retrieval/rephrasers/base.py
diff --git a/...ument_search/retrieval/rephrasers/noop.py → ...ument_search/retrieval/rephrasers/noop.py b/...ument_search/retrieval/rephrasers/noop.py → ...ument_search/retrieval/rephrasers/noop.py
@@ -1,4 +1,4 @@
-from ragbits_document_search.retrieval.rephrasers.base import QueryRephraser
+from ragbits.document_search.retrieval.rephrasers.base import QueryRephraser
 
 
 class NoopQueryRephraser(QueryRephraser):

diff --git a/...nt_search/retrieval/rerankers/__init__.py → ...nt_search/retrieval/rerankers/__init__.py b/...nt_search/retrieval/rerankers/__init__.py → ...nt_search/retrieval/rerankers/__init__.py
diff --git a/...cument_search/retrieval/rerankers/base.py → ...cument_search/retrieval/rerankers/base.py b/...cument_search/retrieval/rerankers/base.py → ...cument_search/retrieval/rerankers/base.py
@@ -1,6 +1,6 @@
 import abc
 
-from ragbits_document_search.documents.element import Element
+from ragbits.document_search.documents.element import Element
 
 
 class Reranker(abc.ABC):

diff --git a/...cument_search/retrieval/rerankers/noop.py → ...cument_search/retrieval/rerankers/noop.py b/...cument_search/retrieval/rerankers/noop.py → ...cument_search/retrieval/rerankers/noop.py
@@ -1,7 +1,7 @@
 from typing import List
 
-from ragbits_document_search.documents.element import Element
-from ragbits_document_search.retrieval.rerankers.base import Reranker
+from ragbits.document_search.documents.element import Element
+from ragbits.document_search.retrieval.rerankers.base import Reranker
 
 
 class NoopReranker(Reranker):

diff --git a/..._document_search/vector_store/__init__.py → .../document_search/vector_store/__init__.py b/..._document_search/vector_store/__init__.py → .../document_search/vector_store/__init__.py
diff --git a/...bits_document_search/vector_store/base.py → ...bits/document_search/vector_store/base.py b/...bits_document_search/vector_store/base.py → ...bits/document_search/vector_store/base.py
diff --git a/...document_search/vector_store/in_memory.py → ...document_search/vector_store/in_memory.py b/...document_search/vector_store/in_memory.py → ...document_search/vector_store/in_memory.py
@@ -1,14 +1,15 @@
 import numpy as np
-from ragbits_document_search.vector_store.base import VectorDBEntry, VectorStore
+
+from ragbits.document_search.vector_store.base import VectorDBEntry, VectorStore
 
 
 class InMemoryVectorStore(VectorStore):
     """
     A simple in-memory implementation of Vector Store, storing vectors in memory.
     """
 
-    def __init__(self):
-        self._storage = {}
+    def __init__(self) -> None:
+        self._storage: dict[str, VectorDBEntry] = {}
 
     async def store(self, entries: list[VectorDBEntry]) -> None:
         """

diff --git a/packages/ragbits-document-search/src/ragbits/py.typed b/packages/ragbits-document-search/src/ragbits/py.typed
diff --git a/packages/ragbits-document-search/tests/unit/test_document_search.py b/packages/ragbits-document-search/tests/unit/test_document_search.py
@@ -1,9 +1,9 @@
 from unittest.mock import AsyncMock
 
-from ragbits_document_search import DocumentSearch
-from ragbits_document_search.documents.document import DocumentMeta
-from ragbits_document_search.documents.element import TextElement
-from ragbits_document_search.vector_store.in_memory import InMemoryVectorStore
+from ragbits.document_search import DocumentSearch
+from ragbits.document_search.documents.document import DocumentMeta
+from ragbits.document_search.documents.element import TextElement
+from ragbits.document_search.vector_store.in_memory import InMemoryVectorStore
 
 
 async def test_document_search():

diff --git a/packages/ragbits-document-search/tests/unit/test_documents.py b/packages/ragbits-document-search/tests/unit/test_documents.py
@@ -1,8 +1,8 @@
 import tempfile
 from pathlib import Path
 
-from ragbits_document_search.documents.document import DocumentMeta, DocumentType, TextDocument
-from ragbits_document_search.documents.sources import LocalFileSource
+from ragbits.document_search.documents.document import DocumentMeta, DocumentType, TextDocument
+from ragbits.document_search.documents.sources import LocalFileSource
 
 
 async def test_loading_local_file_source():

diff --git a/packages/ragbits-document-search/tests/unit/test_elements.py b/packages/ragbits-document-search/tests/unit/test_elements.py
@@ -1,6 +1,6 @@
-from ragbits_document_search.documents.document import DocumentType
-from ragbits_document_search.documents.element import Element
-from ragbits_document_search.vector_store.base import VectorDBEntry
+from ragbits.document_search.documents.document import DocumentType
+from ragbits.document_search.documents.element import Element
+from ragbits.document_search.vector_store.base import VectorDBEntry
 
 
 def test_resolving_element_type():

diff --git a/packages/ragbits-document-search/tests/unit/test_simple_vector_store.py b/packages/ragbits-document-search/tests/unit/test_simple_vector_store.py
@@ -1,9 +1,9 @@
 from pathlib import Path
 
-from ragbits_document_search.documents.document import DocumentMeta, DocumentType
-from ragbits_document_search.documents.element import TextElement
-from ragbits_document_search.documents.sources import LocalFileSource
-from ragbits_document_search.vector_store.in_memory import InMemoryVectorStore
+from ragbits.document_search.documents.document import DocumentMeta, DocumentType
+from ragbits.document_search.documents.element import TextElement
+from ragbits.document_search.documents.sources import LocalFileSource
+from ragbits.document_search.vector_store.in_memory import InMemoryVectorStore
 
 
 async def test_simple_vector_store():