diff --git a/.github/ISSUE_TEMPLATE/01_feature_request.yml b/.github/ISSUE_TEMPLATE/01_feature_request.yml index e53b91e75..5b4ed6bcc 100644 --- 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 index 0b56b812f..56160a65f 100644 --- 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 index 98ab9465e..5b7a1cbf2 100644 --- a/README.md +++ b/README.md @@ -1,4 +1,4 @@ -# Ragnarok +# ragbits Repository for internal experiment with our upcoming LLM framework. diff --git a/packages/ragnarok-common/examples/__init__.py b/packages/ragbits-common/examples/__init__.py similarity index 100% rename from packages/ragnarok-common/examples/__init__.py rename to packages/ragbits-common/examples/__init__.py diff --git a/packages/ragnarok-common/examples/llm_example.py b/packages/ragbits-common/examples/llm_example.py similarity index 93% rename from packages/ragnarok-common/examples/llm_example.py rename to packages/ragbits-common/examples/llm_example.py index 3405d960b..4a2c29bdb 100644 --- a/packages/ragnarok-common/examples/llm_example.py +++ b/packages/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/packages/ragnarok-common/examples/prompt_example.py b/packages/ragbits-common/examples/prompt_example.py similarity index 96% rename from packages/ragnarok-common/examples/prompt_example.py rename to packages/ragbits-common/examples/prompt_example.py index 11e058940..70bcb5e1e 100644 --- a/packages/ragnarok-common/examples/prompt_example.py +++ b/packages/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 b/packages/ragbits-common/pyproject.toml similarity index 100% rename from packages/ragnarok-common/pyproject.toml rename to packages/ragbits-common/pyproject.toml diff --git a/packages/ragnarok-common/setup.cfg b/packages/ragbits-common/setup.cfg similarity index 92% rename from packages/ragnarok-common/setup.cfg rename to packages/ragbits-common/setup.cfg index b8bdeb872..d92bc9475 100644 --- a/packages/ragnarok-common/setup.cfg +++ b/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 = contact@deepsense.ai license = Other/Proprietary License diff --git a/packages/ragnarok-common/src/py.typed b/packages/ragbits-common/src/py.typed similarity index 100% rename from packages/ragnarok-common/src/py.typed rename to packages/ragbits-common/src/py.typed diff --git a/packages/ragnarok-common/src/ragnarok_common/__init__.py b/packages/ragbits-common/src/ragbits_common/__init__.py similarity index 100% rename from packages/ragnarok-common/src/ragnarok_common/__init__.py rename to packages/ragbits-common/src/ragbits_common/__init__.py diff --git a/packages/ragnarok-common/src/ragnarok_common/embeddings/__init__.py b/packages/ragbits-common/src/ragbits_common/embeddings/__init__.py similarity index 100% rename from packages/ragnarok-common/src/ragnarok_common/embeddings/__init__.py rename to packages/ragbits-common/src/ragbits_common/embeddings/__init__.py diff --git a/packages/ragnarok-common/src/ragnarok_common/embeddings/base.py b/packages/ragbits-common/src/ragbits_common/embeddings/base.py similarity index 100% rename from packages/ragnarok-common/src/ragnarok_common/embeddings/base.py rename to packages/ragbits-common/src/ragbits_common/embeddings/base.py diff --git a/packages/ragnarok-common/src/ragnarok_common/embeddings/exceptions.py b/packages/ragbits-common/src/ragbits_common/embeddings/exceptions.py similarity index 100% rename from packages/ragnarok-common/src/ragnarok_common/embeddings/exceptions.py rename to packages/ragbits-common/src/ragbits_common/embeddings/exceptions.py diff --git a/packages/ragnarok-common/src/ragnarok_common/embeddings/litellm.py b/packages/ragbits-common/src/ragbits_common/embeddings/litellm.py similarity index 94% rename from packages/ragnarok-common/src/ragnarok_common/embeddings/litellm.py rename to packages/ragbits-common/src/ragbits_common/embeddings/litellm.py index 4a8654fc5..760141478 100644 --- a/packages/ragnarok-common/src/ragnarok_common/embeddings/litellm.py +++ b/packages/ragbits-common/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/packages/ragnarok-common/src/ragnarok_common/llms/__init__.py b/packages/ragbits-common/src/ragbits_common/llms/__init__.py similarity index 100% rename from packages/ragnarok-common/src/ragnarok_common/llms/__init__.py rename to packages/ragbits-common/src/ragbits_common/llms/__init__.py diff --git a/packages/ragnarok-common/src/ragnarok_common/llms/base.py b/packages/ragbits-common/src/ragbits_common/llms/base.py similarity index 97% rename from packages/ragnarok-common/src/ragnarok_common/llms/base.py rename to packages/ragbits-common/src/ragbits_common/llms/base.py index 4cef332d5..2b628a3ec 100644 --- a/packages/ragnarok-common/src/ragnarok_common/llms/base.py +++ b/packages/ragbits-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/packages/ragnarok-common/src/ragnarok_common/llms/clients/__init__.py b/packages/ragbits-common/src/ragbits_common/llms/clients/__init__.py similarity index 100% rename from packages/ragnarok-common/src/ragnarok_common/llms/clients/__init__.py rename to packages/ragbits-common/src/ragbits_common/llms/clients/__init__.py diff --git a/packages/ragnarok-common/src/ragnarok_common/llms/clients/base.py b/packages/ragbits-common/src/ragbits_common/llms/clients/base.py similarity index 98% rename from packages/ragnarok-common/src/ragnarok_common/llms/clients/base.py rename to packages/ragbits-common/src/ragbits_common/llms/clients/base.py index ae9c5ffbd..2cbd37668 100644 --- a/packages/ragnarok-common/src/ragnarok_common/llms/clients/base.py +++ b/packages/ragbits-common/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/packages/ragnarok-common/src/ragnarok_common/llms/clients/exceptions.py b/packages/ragbits-common/src/ragbits_common/llms/clients/exceptions.py similarity index 100% rename from packages/ragnarok-common/src/ragnarok_common/llms/clients/exceptions.py rename to packages/ragbits-common/src/ragbits_common/llms/clients/exceptions.py diff --git a/packages/ragnarok-common/src/ragnarok_common/llms/clients/litellm.py b/packages/ragbits-common/src/ragbits_common/llms/clients/litellm.py similarity index 99% rename from packages/ragnarok-common/src/ragnarok_common/llms/clients/litellm.py rename to packages/ragbits-common/src/ragbits_common/llms/clients/litellm.py index 0824c55e7..30ff43cae 100644 --- a/packages/ragnarok-common/src/ragnarok_common/llms/clients/litellm.py +++ b/packages/ragbits-common/src/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/packages/ragnarok-common/src/ragnarok_common/llms/clients/local.py b/packages/ragbits-common/src/ragbits_common/llms/clients/local.py similarity index 98% rename from packages/ragnarok-common/src/ragnarok_common/llms/clients/local.py rename to packages/ragbits-common/src/ragbits_common/llms/clients/local.py index 0acb3e9d2..a0941f7a2 100644 --- a/packages/ragnarok-common/src/ragnarok_common/llms/clients/local.py +++ b/packages/ragbits-common/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/packages/ragnarok-common/src/ragnarok_common/llms/litellm.py b/packages/ragbits-common/src/ragbits_common/llms/litellm.py similarity index 98% rename from packages/ragnarok-common/src/ragnarok_common/llms/litellm.py rename to packages/ragbits-common/src/ragbits_common/llms/litellm.py index 2cfebe23e..2c3e1b30c 100644 --- a/packages/ragnarok-common/src/ragnarok_common/llms/litellm.py +++ b/packages/ragbits-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/packages/ragnarok-common/src/ragnarok_common/llms/local.py b/packages/ragbits-common/src/ragbits_common/llms/local.py similarity index 97% rename from packages/ragnarok-common/src/ragnarok_common/llms/local.py rename to packages/ragbits-common/src/ragbits_common/llms/local.py index 482850dc1..9b3000a84 100644 --- a/packages/ragnarok-common/src/ragnarok_common/llms/local.py +++ b/packages/ragbits-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/packages/ragnarok-common/src/ragnarok_common/llms/types.py b/packages/ragbits-common/src/ragbits_common/llms/types.py similarity index 100% rename from packages/ragnarok-common/src/ragnarok_common/llms/types.py rename to packages/ragbits-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 new file mode 100644 index 000000000..8417e3c07 --- /dev/null +++ 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/packages/ragnarok-common/src/ragnarok_common/prompt/base.py b/packages/ragbits-common/src/ragbits_common/prompt/base.py similarity index 100% rename from packages/ragnarok-common/src/ragnarok_common/prompt/base.py rename to packages/ragbits-common/src/ragbits_common/prompt/base.py diff --git a/packages/ragnarok-common/src/ragnarok_common/prompt/parsers.py b/packages/ragbits-common/src/ragbits_common/prompt/parsers.py similarity index 100% rename from packages/ragnarok-common/src/ragnarok_common/prompt/parsers.py rename to packages/ragbits-common/src/ragbits_common/prompt/parsers.py diff --git a/packages/ragnarok-common/src/ragnarok_common/prompt/prompt.py b/packages/ragbits-common/src/ragbits_common/prompt/prompt.py similarity index 100% rename from packages/ragnarok-common/src/ragnarok_common/prompt/prompt.py rename to packages/ragbits-common/src/ragbits_common/prompt/prompt.py diff --git a/packages/ragnarok-common/src/ragnarok_common/py.typed b/packages/ragbits-common/src/ragbits_common/py.typed similarity index 100% rename from packages/ragnarok-common/src/ragnarok_common/py.typed rename to packages/ragbits-common/src/ragbits_common/py.typed diff --git a/packages/ragnarok-common/tests/__init__.py b/packages/ragbits-common/tests/__init__.py similarity index 100% rename from packages/ragnarok-common/tests/__init__.py rename to packages/ragbits-common/tests/__init__.py diff --git a/packages/ragnarok-common/tests/unit/__init__.py b/packages/ragbits-common/tests/unit/__init__.py similarity index 100% rename from packages/ragnarok-common/tests/unit/__init__.py rename to packages/ragbits-common/tests/unit/__init__.py diff --git a/packages/ragnarok-common/tests/unit/llms/__init__.py b/packages/ragbits-common/tests/unit/llms/__init__.py similarity index 100% rename from packages/ragnarok-common/tests/unit/llms/__init__.py rename to packages/ragbits-common/tests/unit/llms/__init__.py diff --git a/packages/ragnarok-common/tests/unit/llms/test_litellm.py b/packages/ragbits-common/tests/unit/llms/test_litellm.py similarity index 94% rename from packages/ragnarok-common/tests/unit/llms/test_litellm.py rename to packages/ragbits-common/tests/unit/llms/test_litellm.py index 284ea22d0..705e0fce2 100644 --- a/packages/ragnarok-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 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/packages/ragnarok-common/tests/unit/prompts/__init__.py b/packages/ragbits-common/tests/unit/prompts/__init__.py similarity index 100% rename from packages/ragnarok-common/tests/unit/prompts/__init__.py rename to packages/ragbits-common/tests/unit/prompts/__init__.py diff --git a/packages/ragnarok-common/tests/unit/prompts/test_parsers.py b/packages/ragbits-common/tests/unit/prompts/test_parsers.py similarity index 97% rename from packages/ragnarok-common/tests/unit/prompts/test_parsers.py rename to packages/ragbits-common/tests/unit/prompts/test_parsers.py index 696f63679..a8302cba4 100644 --- a/packages/ragnarok-common/tests/unit/prompts/test_parsers.py +++ b/packages/ragbits-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/packages/ragnarok-common/tests/unit/prompts/test_prompt.py b/packages/ragbits-common/tests/unit/prompts/test_prompt.py similarity index 99% rename from packages/ragnarok-common/tests/unit/prompts/test_prompt.py rename to packages/ragbits-common/tests/unit/prompts/test_prompt.py index 79e251630..0f4e757c9 100644 --- a/packages/ragnarok-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 ragnarok_common.prompt import Prompt +from ragbits_common.prompt import Prompt class _PromptInput(pydantic.BaseModel): diff --git a/packages/ragnarok-dev-kit/pyproject.toml b/packages/ragbits-dev-kit/pyproject.toml similarity index 100% rename from packages/ragnarok-dev-kit/pyproject.toml rename to packages/ragbits-dev-kit/pyproject.toml diff --git a/packages/ragnarok-dev-kit/setup.cfg b/packages/ragbits-dev-kit/setup.cfg similarity index 93% rename from packages/ragnarok-dev-kit/setup.cfg rename to packages/ragbits-dev-kit/setup.cfg index f2cb7f510..f3ed1ad1b 100644 --- a/packages/ragnarok-dev-kit/setup.cfg +++ b/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 = contact@deepsense.ai license = Other/Proprietary License diff --git a/packages/ragnarok-dev-kit/src/ragnarok_dev_kit/__init__.py b/packages/ragbits-dev-kit/src/ragbits_dev_kit/__init__.py similarity index 100% rename from packages/ragnarok-dev-kit/src/ragnarok_dev_kit/__init__.py rename to packages/ragbits-dev-kit/src/ragbits_dev_kit/__init__.py diff --git a/packages/ragnarok-dev-kit/src/ragnarok_dev_kit/app/__init__.py b/packages/ragbits-dev-kit/src/ragbits_dev_kit/app/__init__.py similarity index 100% rename from packages/ragnarok-dev-kit/src/ragnarok_dev_kit/app/__init__.py rename to packages/ragbits-dev-kit/src/ragbits_dev_kit/app/__init__.py diff --git a/packages/ragnarok-dev-kit/src/ragnarok_dev_kit/app/main.py b/packages/ragbits-dev-kit/src/ragbits_dev_kit/app/main.py similarity index 98% rename from packages/ragnarok-dev-kit/src/ragnarok_dev_kit/app/main.py rename to packages/ragbits-dev-kit/src/ragbits_dev_kit/app/main.py index 292ec9808..d0db70d26 100644 --- a/packages/ragnarok-dev-kit/src/ragnarok_dev_kit/app/main.py +++ b/packages/ragbits-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 new file mode 100644 index 000000000..ee2fecf01 --- /dev/null +++ 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/packages/ragnarok-dev-kit/src/ragnarok_dev_kit/discovery/prompt_discovery.py b/packages/ragbits-dev-kit/src/ragbits_dev_kit/discovery/prompt_discovery.py similarity index 98% rename from packages/ragnarok-dev-kit/src/ragnarok_dev_kit/discovery/prompt_discovery.py rename to packages/ragbits-dev-kit/src/ragbits_dev_kit/discovery/prompt_discovery.py index 5353ac3dd..38be11571 100644 --- a/packages/ragnarok-dev-kit/src/ragnarok_dev_kit/discovery/prompt_discovery.py +++ b/packages/ragbits-dev-kit/src/ragbits_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/packages/ragnarok-dev-kit/tests/unit/discovery/prompt_classes_for_tests.py b/packages/ragbits-dev-kit/tests/unit/discovery/prompt_classes_for_tests.py similarity index 96% rename from packages/ragnarok-dev-kit/tests/unit/discovery/prompt_classes_for_tests.py rename to packages/ragbits-dev-kit/tests/unit/discovery/prompt_classes_for_tests.py index ea0465ecf..5922b3e10 100644 --- a/packages/ragnarok-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 ragnarok_common.prompt import Prompt +from ragbits_common.prompt import Prompt class PromptForTestInput(BaseModel): diff --git a/packages/ragnarok-dev-kit/tests/unit/discovery/ragnarok_tests_pkg_with_prompts/__init__.py b/packages/ragbits-dev-kit/tests/unit/discovery/ragbits_tests_pkg_with_prompts/__init__.py similarity index 100% rename from packages/ragnarok-dev-kit/tests/unit/discovery/ragnarok_tests_pkg_with_prompts/__init__.py rename to packages/ragbits-dev-kit/tests/unit/discovery/ragbits_tests_pkg_with_prompts/__init__.py diff --git a/packages/ragnarok-dev-kit/tests/unit/discovery/ragnarok_tests_pkg_with_prompts/prompts/__init__.py b/packages/ragbits-dev-kit/tests/unit/discovery/ragbits_tests_pkg_with_prompts/prompts/__init__.py similarity index 100% rename from packages/ragnarok-dev-kit/tests/unit/discovery/ragnarok_tests_pkg_with_prompts/prompts/__init__.py rename to packages/ragbits-dev-kit/tests/unit/discovery/ragbits_tests_pkg_with_prompts/prompts/__init__.py diff --git a/packages/ragnarok-dev-kit/tests/unit/discovery/ragnarok_tests_pkg_with_prompts/prompts/temp_prompt1.py b/packages/ragbits-dev-kit/tests/unit/discovery/ragbits_tests_pkg_with_prompts/prompts/temp_prompt1.py similarity index 92% rename from packages/ragnarok-dev-kit/tests/unit/discovery/ragnarok_tests_pkg_with_prompts/prompts/temp_prompt1.py rename to packages/ragbits-dev-kit/tests/unit/discovery/ragbits_tests_pkg_with_prompts/prompts/temp_prompt1.py index bf5c22cb2..0a0c403db 100644 --- a/packages/ragnarok-dev-kit/tests/unit/discovery/ragnarok_tests_pkg_with_prompts/prompts/temp_prompt1.py +++ b/packages/ragbits-dev-kit/tests/unit/discovery/ragbits_tests_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/packages/ragnarok-dev-kit/tests/unit/discovery/ragnarok_tests_pkg_with_prompts/prompts/temp_prompt2.py b/packages/ragbits-dev-kit/tests/unit/discovery/ragbits_tests_pkg_with_prompts/prompts/temp_prompt2.py similarity index 92% rename from packages/ragnarok-dev-kit/tests/unit/discovery/ragnarok_tests_pkg_with_prompts/prompts/temp_prompt2.py rename to packages/ragbits-dev-kit/tests/unit/discovery/ragbits_tests_pkg_with_prompts/prompts/temp_prompt2.py index c756df0ee..99289d337 100644 --- a/packages/ragnarok-dev-kit/tests/unit/discovery/ragnarok_tests_pkg_with_prompts/prompts/temp_prompt2.py +++ b/packages/ragbits-dev-kit/tests/unit/discovery/ragbits_tests_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/packages/ragnarok-dev-kit/tests/unit/discovery/test_prompt_discovery.py b/packages/ragbits-dev-kit/tests/unit/discovery/test_prompt_discovery.py similarity index 91% rename from packages/ragnarok-dev-kit/tests/unit/discovery/test_prompt_discovery.py rename to packages/ragbits-dev-kit/tests/unit/discovery/test_prompt_discovery.py index 4e3cfaf50..0f44ba508 100644 --- a/packages/ragnarok-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 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/packages/ragnarok-document-search/examples/simple_text.py b/packages/ragbits-document-search/examples/simple_text.py similarity index 75% rename from packages/ragnarok-document-search/examples/simple_text.py rename to packages/ragbits-document-search/examples/simple_text.py index 9fc7d2f7f..43bbc96c7 100644 --- a/packages/ragnarok-document-search/examples/simple_text.py +++ b/packages/ragbits-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/packages/ragnarok-document-search/pyproject.toml b/packages/ragbits-document-search/pyproject.toml similarity index 100% rename from packages/ragnarok-document-search/pyproject.toml rename to packages/ragbits-document-search/pyproject.toml diff --git a/packages/ragnarok-document-search/setup.cfg b/packages/ragbits-document-search/setup.cfg similarity index 96% rename from packages/ragnarok-document-search/setup.cfg rename to packages/ragbits-document-search/setup.cfg index 95a0e87b8..a187a55af 100644 --- a/packages/ragnarok-document-search/setup.cfg +++ b/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/packages/ragnarok-document-search/src/py.typed b/packages/ragbits-document-search/src/py.typed similarity index 100% rename from packages/ragnarok-document-search/src/py.typed rename to packages/ragbits-document-search/src/py.typed diff --git a/packages/ragnarok-document-search/src/ragnarok_document_search/__init__.py b/packages/ragbits-document-search/src/ragbits_document_search/__init__.py similarity index 100% rename from packages/ragnarok-document-search/src/ragnarok_document_search/__init__.py rename to packages/ragbits-document-search/src/ragbits_document_search/__init__.py diff --git a/packages/ragnarok-document-search/src/ragnarok_document_search/__version__.py b/packages/ragbits-document-search/src/ragbits_document_search/__version__.py similarity index 100% rename from packages/ragnarok-document-search/src/ragnarok_document_search/__version__.py rename to packages/ragbits-document-search/src/ragbits_document_search/__version__.py diff --git a/packages/ragnarok-document-search/src/ragnarok_document_search/_main.py b/packages/ragbits-document-search/src/ragbits_document_search/_main.py similarity index 77% rename from packages/ragnarok-document-search/src/ragnarok_document_search/_main.py rename to packages/ragbits-document-search/src/ragbits_document_search/_main.py index a1642668d..a0a1257d9 100644 --- a/packages/ragnarok-document-search/src/ragnarok_document_search/_main.py +++ b/packages/ragbits-document-search/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/packages/ragnarok-document-search/src/ragnarok_document_search/documents/__init__.py b/packages/ragbits-document-search/src/ragbits_document_search/documents/__init__.py similarity index 100% rename from packages/ragnarok-document-search/src/ragnarok_document_search/documents/__init__.py rename to packages/ragbits-document-search/src/ragbits_document_search/documents/__init__.py diff --git a/packages/ragnarok-document-search/src/ragnarok_document_search/documents/document.py b/packages/ragbits-document-search/src/ragbits_document_search/documents/document.py similarity index 97% rename from packages/ragnarok-document-search/src/ragnarok_document_search/documents/document.py rename to packages/ragbits-document-search/src/ragbits_document_search/documents/document.py index 7810b719d..e6a551056 100644 --- a/packages/ragnarok-document-search/src/ragnarok_document_search/documents/document.py +++ b/packages/ragbits-document-search/src/ragbits_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/packages/ragnarok-document-search/src/ragnarok_document_search/documents/element.py b/packages/ragbits-document-search/src/ragbits_document_search/documents/element.py similarity index 93% rename from packages/ragnarok-document-search/src/ragnarok_document_search/documents/element.py rename to packages/ragbits-document-search/src/ragbits_document_search/documents/element.py index ea1e4aee3..596e4af1f 100644 --- a/packages/ragnarok-document-search/src/ragnarok_document_search/documents/element.py +++ b/packages/ragbits-document-search/src/ragbits_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/packages/ragnarok-document-search/src/ragnarok_document_search/documents/sources.py b/packages/ragbits-document-search/src/ragbits_document_search/documents/sources.py similarity index 100% rename from packages/ragnarok-document-search/src/ragnarok_document_search/documents/sources.py rename to packages/ragbits-document-search/src/ragbits_document_search/documents/sources.py diff --git a/packages/ragnarok-document-search/src/ragnarok_document_search/ingestion/__init__.py b/packages/ragbits-document-search/src/ragbits_document_search/ingestion/__init__.py similarity index 100% rename from packages/ragnarok-document-search/src/ragnarok_document_search/ingestion/__init__.py rename to packages/ragbits-document-search/src/ragbits_document_search/ingestion/__init__.py diff --git a/packages/ragnarok-document-search/src/ragnarok_document_search/ingestion/document_processor.py b/packages/ragbits-document-search/src/ragbits_document_search/ingestion/document_processor.py similarity index 85% rename from packages/ragnarok-document-search/src/ragnarok_document_search/ingestion/document_processor.py rename to packages/ragbits-document-search/src/ragbits_document_search/ingestion/document_processor.py index 5f10301fa..8e445552d 100644 --- a/packages/ragnarok-document-search/src/ragnarok_document_search/ingestion/document_processor.py +++ b/packages/ragbits-document-search/src/ragbits_document_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/packages/ragnarok-document-search/src/ragnarok_document_search/py.typed b/packages/ragbits-document-search/src/ragbits_document_search/py.typed similarity index 100% rename from packages/ragnarok-document-search/src/ragnarok_document_search/py.typed rename to packages/ragbits-document-search/src/ragbits_document_search/py.typed diff --git a/packages/ragnarok-document-search/src/ragnarok_document_search/retrieval/__init__.py b/packages/ragbits-document-search/src/ragbits_document_search/retrieval/__init__.py similarity index 100% rename from packages/ragnarok-document-search/src/ragnarok_document_search/retrieval/__init__.py rename to packages/ragbits-document-search/src/ragbits_document_search/retrieval/__init__.py diff --git a/packages/ragnarok-document-search/src/ragnarok_document_search/retrieval/rephrasers/__init__.py b/packages/ragbits-document-search/src/ragbits_document_search/retrieval/rephrasers/__init__.py similarity index 100% rename from packages/ragnarok-document-search/src/ragnarok_document_search/retrieval/rephrasers/__init__.py rename to packages/ragbits-document-search/src/ragbits_document_search/retrieval/rephrasers/__init__.py diff --git a/packages/ragnarok-document-search/src/ragnarok_document_search/retrieval/rephrasers/base.py b/packages/ragbits-document-search/src/ragbits_document_search/retrieval/rephrasers/base.py similarity index 100% rename from packages/ragnarok-document-search/src/ragnarok_document_search/retrieval/rephrasers/base.py rename to packages/ragbits-document-search/src/ragbits_document_search/retrieval/rephrasers/base.py diff --git a/packages/ragnarok-document-search/src/ragnarok_document_search/retrieval/rephrasers/noop.py b/packages/ragbits-document-search/src/ragbits_document_search/retrieval/rephrasers/noop.py similarity index 84% rename from packages/ragnarok-document-search/src/ragnarok_document_search/retrieval/rephrasers/noop.py rename to packages/ragbits-document-search/src/ragbits_document_search/retrieval/rephrasers/noop.py index f89ab39bb..30e88e33d 100644 --- a/packages/ragnarok-document-search/src/ragnarok_document_search/retrieval/rephrasers/noop.py +++ b/packages/ragbits-document-search/src/ragbits_document_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/packages/ragnarok-document-search/src/ragnarok_document_search/retrieval/rerankers/__init__.py b/packages/ragbits-document-search/src/ragbits_document_search/retrieval/rerankers/__init__.py similarity index 100% rename from packages/ragnarok-document-search/src/ragnarok_document_search/retrieval/rerankers/__init__.py rename to packages/ragbits-document-search/src/ragbits_document_search/retrieval/rerankers/__init__.py diff --git a/packages/ragnarok-document-search/src/ragnarok_document_search/retrieval/rerankers/base.py b/packages/ragbits-document-search/src/ragbits_document_search/retrieval/rerankers/base.py similarity index 85% rename from packages/ragnarok-document-search/src/ragnarok_document_search/retrieval/rerankers/base.py rename to packages/ragbits-document-search/src/ragbits_document_search/retrieval/rerankers/base.py index 2c941d999..2ddffde5c 100644 --- a/packages/ragnarok-document-search/src/ragnarok_document_search/retrieval/rerankers/base.py +++ b/packages/ragbits-document-search/src/ragbits_document_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/packages/ragnarok-document-search/src/ragnarok_document_search/retrieval/rerankers/noop.py b/packages/ragbits-document-search/src/ragbits_document_search/retrieval/rerankers/noop.py similarity index 76% rename from packages/ragnarok-document-search/src/ragnarok_document_search/retrieval/rerankers/noop.py rename to packages/ragbits-document-search/src/ragbits_document_search/retrieval/rerankers/noop.py index 9ef73cbb6..1e943fcd6 100644 --- a/packages/ragnarok-document-search/src/ragnarok_document_search/retrieval/rerankers/noop.py +++ b/packages/ragbits-document-search/src/ragbits_document_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/packages/ragnarok-document-search/src/ragnarok_document_search/vector_store/__init__.py b/packages/ragbits-document-search/src/ragbits_document_search/vector_store/__init__.py similarity index 100% rename from packages/ragnarok-document-search/src/ragnarok_document_search/vector_store/__init__.py rename to packages/ragbits-document-search/src/ragbits_document_search/vector_store/__init__.py diff --git a/packages/ragnarok-document-search/src/ragnarok_document_search/vector_store/base.py b/packages/ragbits-document-search/src/ragbits_document_search/vector_store/base.py similarity index 100% rename from packages/ragnarok-document-search/src/ragnarok_document_search/vector_store/base.py rename to packages/ragbits-document-search/src/ragbits_document_search/vector_store/base.py diff --git a/packages/ragnarok-document-search/src/ragnarok_document_search/vector_store/in_memory.py b/packages/ragbits-document-search/src/ragbits_document_search/vector_store/in_memory.py similarity index 93% rename from packages/ragnarok-document-search/src/ragnarok_document_search/vector_store/in_memory.py rename to packages/ragbits-document-search/src/ragbits_document_search/vector_store/in_memory.py index babe6b1c4..2882fa1a9 100644 --- a/packages/ragnarok-document-search/src/ragnarok_document_search/vector_store/in_memory.py +++ b/packages/ragbits-document-search/src/ragbits_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/packages/ragnarok-document-search/tests/unit/__init__.py b/packages/ragbits-document-search/tests/unit/__init__.py similarity index 100% rename from packages/ragnarok-document-search/tests/unit/__init__.py rename to packages/ragbits-document-search/tests/unit/__init__.py diff --git a/packages/ragnarok-document-search/tests/unit/test_document_search.py b/packages/ragbits-document-search/tests/unit/test_document_search.py similarity index 69% rename from packages/ragnarok-document-search/tests/unit/test_document_search.py rename to packages/ragbits-document-search/tests/unit/test_document_search.py index 6d8b3bbbe..8bd5183ee 100644 --- a/packages/ragnarok-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 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/packages/ragnarok-document-search/tests/unit/test_documents.py b/packages/ragbits-document-search/tests/unit/test_documents.py similarity index 73% rename from packages/ragnarok-document-search/tests/unit/test_documents.py rename to packages/ragbits-document-search/tests/unit/test_documents.py index beb2ed6aa..c293cb999 100644 --- a/packages/ragnarok-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 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/packages/ragnarok-document-search/tests/unit/test_elements.py b/packages/ragbits-document-search/tests/unit/test_elements.py similarity index 80% rename from packages/ragnarok-document-search/tests/unit/test_elements.py rename to packages/ragbits-document-search/tests/unit/test_elements.py index 05c73a324..8a04348c5 100644 --- a/packages/ragnarok-document-search/tests/unit/test_elements.py +++ b/packages/ragbits-document-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/packages/ragnarok-document-search/tests/unit/test_simple_vector_store.py b/packages/ragbits-document-search/tests/unit/test_simple_vector_store.py similarity index 70% rename from packages/ragnarok-document-search/tests/unit/test_simple_vector_store.py rename to packages/ragbits-document-search/tests/unit/test_simple_vector_store.py index 415106832..177eb6fa8 100644 --- a/packages/ragnarok-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 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(): diff --git a/packages/ragnarok-common/src/ragnarok_common/prompt/__init__.py b/packages/ragnarok-common/src/ragnarok_common/prompt/__init__.py deleted file mode 100644 index ac4542a8d..000000000 --- a/packages/ragnarok-common/src/ragnarok_common/prompt/__init__.py +++ /dev/null @@ -1,3 +0,0 @@ -from ragnarok_common.prompt.prompt import ChatFormat, Prompt - -__all__ = ["Prompt", "ChatFormat"] diff --git a/packages/ragnarok-dev-kit/src/ragnarok_dev_kit/discovery/__init__.py b/packages/ragnarok-dev-kit/src/ragnarok_dev_kit/discovery/__init__.py deleted file mode 100644 index a6200bbac..000000000 --- a/packages/ragnarok-dev-kit/src/ragnarok_dev_kit/discovery/__init__.py +++ /dev/null @@ -1,3 +0,0 @@ -from ragnarok_dev_kit.discovery.prompt_discovery import PromptDiscovery - -__all__ = ["PromptDiscovery"] diff --git a/pyproject.toml b/pyproject.toml index 1c16ed536..a5cabe1f6 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,9 +1,8 @@ - [tool.isort] multi_line_output = 3 line_length = 120 include_trailing_comma = true -known_first_party = ['ragnarok_common'] +known_first_party = ['ragbits_common'] known_third_party = [ # Most popular libraries. Extend if necessary. 'IPython', 'PIL', @@ -85,7 +84,7 @@ no_implicit_optional = true mypy_path = ['packages'] [[tool.mypy.overrides]] -module = "ragnarok_common.*" +module = "ragbits_common.*" ignore_missing_imports = false disallow_untyped_defs = true diff --git a/requirements-dev.txt b/requirements-dev.txt index 8df9981af..c1a2e6f0d 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -10,4 +10,4 @@ pip-licenses>=4.0.0,<5.0.0 gradio==4.42.0 # extras required by tests: --e packages/ragnarok-common[litellm] +-e packages/ragbits-common[litellm]