chore: rename project to ragbits

deepsense-ai · Sep 16, 2024 · a44f97d · a44f97d
1 parent c8ae2f9
commit a44f97d
Show file tree

Hide file tree

Showing 82 changed files with 82 additions and 83 deletions.
diff --git a/.github/ISSUE_TEMPLATE/01_feature_request.yml b/.github/ISSUE_TEMPLATE/01_feature_request.yml
@@ -1,12 +1,12 @@
 name: 🚀 Feature Request
-description: Submit a proposal/request for a new ragnarok feature.
+description: Submit a proposal/request for a new ragbits feature.
 title: "feat: "
 labels: ["feature"]
 body:
   - type: markdown
     attributes:
       value: |
-        Thanks for contributing to ragnarok!
+        Thanks for contributing to ragbits!
   - type: textarea
     id: feature-description
     attributes:

diff --git a/.github/ISSUE_TEMPLATE/02_bug_report.yml b/.github/ISSUE_TEMPLATE/02_bug_report.yml
@@ -21,7 +21,7 @@ body:
     attributes:
       label: How can we reproduce it?
       description: Please provide a code snippet to reproduce the bug.
-      placeholder: import ragnarok
+      placeholder: import ragbits
       render: python
   - type: textarea
     id: logs

diff --git a/README.md b/README.md
@@ -1,4 +1,4 @@
-# Ragnarok
+# ragbits
 
 Repository for internal experiment with our upcoming LLM framework.
 

diff --git a/...ages/ragnarok-common/examples/__init__.py → packages/ragbits-common/examples/__init__.py b/...ages/ragnarok-common/examples/__init__.py → packages/ragbits-common/examples/__init__.py
diff --git a/...s/ragnarok-common/examples/llm_example.py → ...es/ragbits-common/examples/llm_example.py b/...s/ragnarok-common/examples/llm_example.py → ...es/ragbits-common/examples/llm_example.py
@@ -2,8 +2,8 @@
 
 from pydantic import BaseModel
 
-from ragnarok_common.llms.litellm import LiteLLM
-from ragnarok_common.prompt import Prompt
+from ragbits_common.llms.litellm import LiteLLM
+from ragbits_common.prompt import Prompt
 
 
 class LoremPromptInput(BaseModel):

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

diff --git a/packages/ragnarok-common/pyproject.toml → packages/ragbits-common/pyproject.toml b/packages/ragnarok-common/pyproject.toml → packages/ragbits-common/pyproject.toml
diff --git a/packages/ragnarok-common/setup.cfg → packages/ragbits-common/setup.cfg b/packages/ragnarok-common/setup.cfg → packages/ragbits-common/setup.cfg
@@ -1,7 +1,7 @@
 [metadata]
-name = ragnarok-common
+name = ragbits-common
 version = 0.0.1
-description = "common utilities for the ragnarok framework"
+description = "common utilities for the ragbits framework"
 author = deepsense.ai
 author_email = [email protected]
 license = Other/Proprietary License

diff --git a/packages/ragnarok-common/src/py.typed → packages/ragbits-common/src/py.typed b/packages/ragnarok-common/src/py.typed → packages/ragbits-common/src/py.typed
diff --git a/...ok-common/src/ragnarok_common/__init__.py → ...its-common/src/ragbits_common/__init__.py b/...ok-common/src/ragnarok_common/__init__.py → ...its-common/src/ragbits_common/__init__.py
diff --git a/...rc/ragnarok_common/embeddings/__init__.py → ...src/ragbits_common/embeddings/__init__.py b/...rc/ragnarok_common/embeddings/__init__.py → ...src/ragbits_common/embeddings/__init__.py
diff --git a/...on/src/ragnarok_common/embeddings/base.py → ...mon/src/ragbits_common/embeddings/base.py b/...on/src/ragnarok_common/embeddings/base.py → ...mon/src/ragbits_common/embeddings/base.py
diff --git a/.../ragnarok_common/embeddings/exceptions.py → ...c/ragbits_common/embeddings/exceptions.py b/.../ragnarok_common/embeddings/exceptions.py → ...c/ragbits_common/embeddings/exceptions.py
diff --git a/...src/ragnarok_common/embeddings/litellm.py → .../src/ragbits_common/embeddings/litellm.py b/...src/ragnarok_common/embeddings/litellm.py → .../src/ragbits_common/embeddings/litellm.py
@@ -7,8 +7,8 @@
 except ImportError:
     HAS_LITELLM = False
 
-from ragnarok_common.embeddings.base import Embeddings
-from ragnarok_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/...mmon/src/ragnarok_common/llms/__init__.py → ...ommon/src/ragbits_common/llms/__init__.py b/...mmon/src/ragnarok_common/llms/__init__.py → ...ommon/src/ragbits_common/llms/__init__.py
diff --git a/...k-common/src/ragnarok_common/llms/base.py → ...ts-common/src/ragbits_common/llms/base.py b/...k-common/src/ragnarok_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 ragnarok_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/.../ragnarok_common/llms/clients/__init__.py → ...c/ragbits_common/llms/clients/__init__.py b/.../ragnarok_common/llms/clients/__init__.py → ...c/ragbits_common/llms/clients/__init__.py
diff --git a/.../src/ragnarok_common/llms/clients/base.py → ...n/src/ragbits_common/llms/clients/base.py b/.../src/ragnarok_common/llms/clients/base.py → ...n/src/ragbits_common/llms/clients/base.py
@@ -4,7 +4,7 @@
 
 from pydantic import BaseModel
 
-from ragnarok_common.prompt import ChatFormat
+from ragbits_common.prompt import ChatFormat
 
 from ..types import NotGiven
 

diff --git a/...agnarok_common/llms/clients/exceptions.py → ...ragbits_common/llms/clients/exceptions.py b/...agnarok_common/llms/clients/exceptions.py → ...ragbits_common/llms/clients/exceptions.py
diff --git a/...c/ragnarok_common/llms/clients/litellm.py → ...rc/ragbits_common/llms/clients/litellm.py b/...c/ragnarok_common/llms/clients/litellm.py → ...rc/ragbits_common/llms/clients/litellm.py
@@ -11,7 +11,7 @@
     HAS_LITELLM = False
 
 
-from ragnarok_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/ragnarok_common/llms/clients/local.py → .../src/ragbits_common/llms/clients/local.py b/...src/ragnarok_common/llms/clients/local.py → .../src/ragbits_common/llms/clients/local.py
@@ -11,7 +11,7 @@
 except ImportError:
     HAS_LOCAL_LLM = False
 
-from ragnarok_common.prompt import ChatFormat
+from ragbits_common.prompt import ChatFormat
 
 from ..types import NOT_GIVEN, NotGiven
 from .base import LLMClient, LLMOptions

diff --git a/...ommon/src/ragnarok_common/llms/litellm.py → ...common/src/ragbits_common/llms/litellm.py b/...ommon/src/ragnarok_common/llms/litellm.py → ...common/src/ragbits_common/llms/litellm.py
@@ -8,7 +8,7 @@
 except ImportError:
     HAS_LITELLM = False
 
-from ragnarok_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/...-common/src/ragnarok_common/llms/local.py → ...s-common/src/ragbits_common/llms/local.py b/...-common/src/ragnarok_common/llms/local.py → ...s-common/src/ragbits_common/llms/local.py
@@ -8,7 +8,7 @@
 except ImportError:
     HAS_LOCAL_LLM = False
 
-from ragnarok_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/...-common/src/ragnarok_common/llms/types.py → ...s-common/src/ragbits_common/llms/types.py b/...-common/src/ragnarok_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/ragnarok_common/prompt/base.py → ...-common/src/ragbits_common/prompt/base.py b/...common/src/ragnarok_common/prompt/base.py → ...-common/src/ragbits_common/prompt/base.py
diff --git a/...mon/src/ragnarok_common/prompt/parsers.py → ...mmon/src/ragbits_common/prompt/parsers.py b/...mon/src/ragnarok_common/prompt/parsers.py → ...mmon/src/ragbits_common/prompt/parsers.py
diff --git a/...mmon/src/ragnarok_common/prompt/prompt.py → ...ommon/src/ragbits_common/prompt/prompt.py b/...mmon/src/ragnarok_common/prompt/prompt.py → ...ommon/src/ragbits_common/prompt/prompt.py
diff --git a/...narok-common/src/ragnarok_common/py.typed → ...agbits-common/src/ragbits_common/py.typed b/...narok-common/src/ragnarok_common/py.typed → ...agbits-common/src/ragbits_common/py.typed
diff --git a/packages/ragnarok-common/tests/__init__.py → packages/ragbits-common/tests/__init__.py b/packages/ragnarok-common/tests/__init__.py → packages/ragbits-common/tests/__init__.py
diff --git a/...es/ragnarok-common/tests/unit/__init__.py → ...ges/ragbits-common/tests/unit/__init__.py b/...es/ragnarok-common/tests/unit/__init__.py → ...ges/ragbits-common/tests/unit/__init__.py
diff --git a/...gnarok-common/tests/unit/llms/__init__.py → ...agbits-common/tests/unit/llms/__init__.py b/...gnarok-common/tests/unit/llms/__init__.py → ...agbits-common/tests/unit/llms/__init__.py
diff --git a/...ok-common/tests/unit/llms/test_litellm.py → ...ts-common/tests/unit/llms/test_litellm.py b/...ok-common/tests/unit/llms/test_litellm.py → ...ts-common/tests/unit/llms/test_litellm.py
@@ -1,9 +1,9 @@
 from pydantic import BaseModel
 
-from ragnarok_common.llms.clients.litellm import LiteLLMOptions
-from ragnarok_common.llms.litellm import LiteLLM
-from ragnarok_common.prompt import Prompt
-from ragnarok_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/...rok-common/tests/unit/prompts/__init__.py → ...its-common/tests/unit/prompts/__init__.py b/...rok-common/tests/unit/prompts/__init__.py → ...its-common/tests/unit/prompts/__init__.py
diff --git a/...common/tests/unit/prompts/test_parsers.py → ...common/tests/unit/prompts/test_parsers.py b/...common/tests/unit/prompts/test_parsers.py → ...common/tests/unit/prompts/test_parsers.py
@@ -2,8 +2,8 @@
 
 import pytest
 
-from ragnarok_common.prompt import Prompt
-from ragnarok_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/...-common/tests/unit/prompts/test_prompt.py → ...-common/tests/unit/prompts/test_prompt.py b/...-common/tests/unit/prompts/test_prompt.py → ...-common/tests/unit/prompts/test_prompt.py
@@ -1,7 +1,7 @@
 import pydantic
 import pytest
 
-from ragnarok_common.prompt import Prompt
+from ragbits_common.prompt import Prompt
 
 
 class _PromptInput(pydantic.BaseModel):

diff --git a/packages/ragnarok-dev-kit/pyproject.toml → packages/ragbits-dev-kit/pyproject.toml b/packages/ragnarok-dev-kit/pyproject.toml → packages/ragbits-dev-kit/pyproject.toml
diff --git a/packages/ragnarok-dev-kit/setup.cfg → packages/ragbits-dev-kit/setup.cfg b/packages/ragnarok-dev-kit/setup.cfg → packages/ragbits-dev-kit/setup.cfg
@@ -1,7 +1,7 @@
 [metadata]
-name = ragnarok-dev-kit
+name = ragbits-dev-kit
 version = 0.0.1
-description = "dev kit for ragnarok framework"
+description = "dev kit for ragbits framework"
 author = deepsense.ai
 author_email = [email protected]
 license = Other/Proprietary License

diff --git a/...-dev-kit/src/ragnarok_dev_kit/__init__.py → ...s-dev-kit/src/ragbits_dev_kit/__init__.py b/...-dev-kit/src/ragnarok_dev_kit/__init__.py → ...s-dev-kit/src/ragbits_dev_kit/__init__.py
diff --git a/...-kit/src/ragnarok_dev_kit/app/__init__.py → ...v-kit/src/ragbits_dev_kit/app/__init__.py b/...-kit/src/ragnarok_dev_kit/app/__init__.py → ...v-kit/src/ragbits_dev_kit/app/__init__.py
diff --git a/...-dev-kit/src/ragnarok_dev_kit/app/main.py → ...s-dev-kit/src/ragbits_dev_kit/app/main.py b/...-dev-kit/src/ragnarok_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 ragnarok_dev_kit.discovery.prompt_discovery import PromptDiscovery
+from ragbits_dev_kit.discovery.prompt_discovery import PromptDiscovery
 
-from ragnarok_common.llms import LiteLLM
-from ragnarok_common.llms.clients import LiteLLMOptions
+from ragbits_common.llms import LiteLLM
+from ragbits_common.llms.clients import LiteLLMOptions
 
 
 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/...rok_dev_kit/discovery/prompt_discovery.py → ...its_dev_kit/discovery/prompt_discovery.py b/...rok_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 ragnarok_common.prompt import Prompt
+from ragbits_common.prompt import Prompt
 
 PromptDetails = namedtuple("PromptDetails", ["system_prompt", "user_prompt", "input_type", "object"])
 

diff --git a/...nit/discovery/prompt_classes_for_tests.py → ...nit/discovery/prompt_classes_for_tests.py b/...nit/discovery/prompt_classes_for_tests.py → ...nit/discovery/prompt_classes_for_tests.py
@@ -2,7 +2,7 @@
 
 from pydantic import BaseModel
 
-from ragnarok_common.prompt import Prompt
+from ragbits_common.prompt import Prompt
 
 
 class PromptForTestInput(BaseModel):

diff --git a/...gnarok_tests_pkg_with_prompts/__init__.py → ...agbits_tests_pkg_with_prompts/__init__.py b/...gnarok_tests_pkg_with_prompts/__init__.py → ...agbits_tests_pkg_with_prompts/__init__.py
diff --git a/...ests_pkg_with_prompts/prompts/__init__.py → ...ests_pkg_with_prompts/prompts/__init__.py b/...ests_pkg_with_prompts/prompts/__init__.py → ...ests_pkg_with_prompts/prompts/__init__.py
diff --git a/..._pkg_with_prompts/prompts/temp_prompt1.py → ..._pkg_with_prompts/prompts/temp_prompt1.py b/..._pkg_with_prompts/prompts/temp_prompt1.py → ..._pkg_with_prompts/prompts/temp_prompt1.py
@@ -1,6 +1,6 @@
 from pydantic import BaseModel
 
-from ragnarok_common.prompt import Prompt
+from ragbits_common.prompt import Prompt
 
 
 class PromptForTestInputA(BaseModel):

diff --git a/..._pkg_with_prompts/prompts/temp_prompt2.py → ..._pkg_with_prompts/prompts/temp_prompt2.py b/..._pkg_with_prompts/prompts/temp_prompt2.py → ..._pkg_with_prompts/prompts/temp_prompt2.py
@@ -1,6 +1,6 @@
 from pydantic import BaseModel
 
-from ragnarok_common.prompt import Prompt
+from ragbits_common.prompt import Prompt
 
 
 class PromptForTestInputB(BaseModel):

diff --git a/...s/unit/discovery/test_prompt_discovery.py → ...s/unit/discovery/test_prompt_discovery.py b/...s/unit/discovery/test_prompt_discovery.py → ...s/unit/discovery/test_prompt_discovery.py
@@ -1,7 +1,7 @@
 import sys
 from pathlib import Path
 
-from ragnarok_dev_kit.discovery.prompt_discovery import PromptDiscovery
+from ragbits_dev_kit.discovery.prompt_discovery import PromptDiscovery
 
 
 def test_prompt_discovery_from_file():
@@ -28,7 +28,7 @@ def test_prompt_discovery_from_file():
 
 def test_prompt_discovery_from_package():
     sys.path.append(str(Path(__file__).parent))
-    test_paths = ["ragnarok_tests_pkg_with_prompts"]
+    test_paths = ["ragbits_tests_pkg_with_prompts"]
 
     discovery_result = PromptDiscovery(test_paths).discover()
 

diff --git a/...k-document-search/examples/simple_text.py → ...s-document-search/examples/simple_text.py b/...k-document-search/examples/simple_text.py → ...s-document-search/examples/simple_text.py
@@ -1,10 +1,10 @@
 import asyncio
 
-from ragnarok_document_search import DocumentSearch
-from ragnarok_document_search.documents.document import DocumentMeta
-from ragnarok_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.vector_store.in_memory import InMemoryVectorStore
 
-from ragnarok_common.embeddings.litellm import LiteLLMEmbeddings
+from ragbits_common.embeddings.litellm import LiteLLMEmbeddings
 
 documents = [
     DocumentMeta.create_text_document_from_literal("RIP boiled water. You will be mist."),

diff --git a/...s/ragnarok-document-search/pyproject.toml → ...es/ragbits-document-search/pyproject.toml b/...s/ragnarok-document-search/pyproject.toml → ...es/ragbits-document-search/pyproject.toml
diff --git a/packages/ragnarok-document-search/setup.cfg → packages/ragbits-document-search/setup.cfg b/packages/ragnarok-document-search/setup.cfg → packages/ragbits-document-search/setup.cfg
@@ -1,5 +1,5 @@
 [metadata]
-name = ragnarok-document-search
+name = ragbits-document-search
 # do not change version by hand: use bump_version.sh
 version = 0.0.1
 description = "The ragstack module responsible for fetching data from unstructured data sources."

diff --git a/...ges/ragnarok-document-search/src/py.typed → ...ages/ragbits-document-search/src/py.typed b/...ges/ragnarok-document-search/src/py.typed → ...ages/ragbits-document-search/src/py.typed
diff --git a/.../src/ragnarok_document_search/__init__.py → ...h/src/ragbits_document_search/__init__.py b/.../src/ragnarok_document_search/__init__.py → ...h/src/ragbits_document_search/__init__.py
diff --git a/...c/ragnarok_document_search/__version__.py → ...rc/ragbits_document_search/__version__.py b/...c/ragnarok_document_search/__version__.py → ...rc/ragbits_document_search/__version__.py
diff --git a/...rch/src/ragnarok_document_search/_main.py → ...arch/src/ragbits_document_search/_main.py b/...rch/src/ragnarok_document_search/_main.py → ...arch/src/ragbits_document_search/_main.py
@@ -1,13 +1,13 @@
-from ragnarok_document_search.documents.document import DocumentMeta
-from ragnarok_document_search.documents.element import Element
-from ragnarok_document_search.ingestion.document_processor import DocumentProcessor
-from ragnarok_document_search.retrieval.rephrasers.base import QueryRephraser
-from ragnarok_document_search.retrieval.rephrasers.noop import NoopQueryRephraser
-from ragnarok_document_search.retrieval.rerankers.base import Reranker
-from ragnarok_document_search.retrieval.rerankers.noop import NoopReranker
-from ragnarok_document_search.vector_store.base import VectorStore
-
-from ragnarok_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
+
+from ragbits_common.embeddings.base import Embeddings
 
 
 class DocumentSearch:

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

diff --git a/...arok_document_search/documents/element.py → ...bits_document_search/documents/element.py b/...arok_document_search/documents/element.py → ...bits_document_search/documents/element.py
@@ -2,8 +2,8 @@
 from typing import ClassVar
 
 from pydantic import BaseModel
-from ragnarok_document_search.documents.document import DocumentMeta
-from ragnarok_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):

diff --git a/...arok_document_search/documents/sources.py → ...bits_document_search/documents/sources.py b/...arok_document_search/documents/sources.py → ...bits_document_search/documents/sources.py
diff --git a/...rok_document_search/ingestion/__init__.py → ...its_document_search/ingestion/__init__.py b/...rok_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 ragnarok_document_search.documents.document import DocumentMeta, TextDocument
-from ragnarok_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/...rch/src/ragnarok_document_search/py.typed → ...arch/src/ragbits_document_search/py.typed b/...rch/src/ragnarok_document_search/py.typed → ...arch/src/ragbits_document_search/py.typed
diff --git a/...rok_document_search/retrieval/__init__.py → ...its_document_search/retrieval/__init__.py b/...rok_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 ragnarok_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 ragnarok_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 ragnarok_document_search.documents.element import Element
-from ragnarok_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/...arok_document_search/vector_store/base.py → ...bits_document_search/vector_store/base.py b/...arok_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,5 +1,5 @@
 import numpy as np
-from ragnarok_document_search.vector_store.base import VectorDBEntry, VectorStore
+from ragbits_document_search.vector_store.base import VectorDBEntry, VectorStore
 
 
 class InMemoryVectorStore(VectorStore):

diff --git a/...ok-document-search/tests/unit/__init__.py → ...ts-document-search/tests/unit/__init__.py b/...ok-document-search/tests/unit/__init__.py → ...ts-document-search/tests/unit/__init__.py
diff --git a/...search/tests/unit/test_document_search.py → ...search/tests/unit/test_document_search.py b/...search/tests/unit/test_document_search.py → ...search/tests/unit/test_document_search.py
@@ -1,9 +1,9 @@
 from unittest.mock import AsyncMock
 
-from ragnarok_document_search import DocumentSearch
-from ragnarok_document_search.documents.document import DocumentMeta
-from ragnarok_document_search.documents.element import TextElement
-from ragnarok_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/...ument-search/tests/unit/test_documents.py → ...ument-search/tests/unit/test_documents.py b/...ument-search/tests/unit/test_documents.py → ...ument-search/tests/unit/test_documents.py
@@ -1,8 +1,8 @@
 import tempfile
 from pathlib import Path
 
-from ragnarok_document_search.documents.document import DocumentMeta, DocumentType, TextDocument
-from ragnarok_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/...cument-search/tests/unit/test_elements.py → ...cument-search/tests/unit/test_elements.py b/...cument-search/tests/unit/test_elements.py → ...cument-search/tests/unit/test_elements.py
@@ -1,6 +1,6 @@
-from ragnarok_document_search.documents.document import DocumentType
-from ragnarok_document_search.documents.element import Element
-from ragnarok_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/...ch/tests/unit/test_simple_vector_store.py → ...ch/tests/unit/test_simple_vector_store.py b/...ch/tests/unit/test_simple_vector_store.py → ...ch/tests/unit/test_simple_vector_store.py
@@ -1,9 +1,9 @@
 from pathlib import Path
 
-from ragnarok_document_search.documents.document import DocumentMeta, DocumentType
-from ragnarok_document_search.documents.element import TextElement
-from ragnarok_document_search.documents.sources import LocalFileSource
-from ragnarok_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():
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		from ragbits_common.prompt.prompt import ChatFormat, Prompt

		__all__ = ["Prompt", "ChatFormat"]
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		from ragbits_dev_kit.discovery.prompt_discovery import PromptDiscovery

		__all__ = ["PromptDiscovery"]