Skip to content

Commit

Permalink
chore: use ragnarok as a project name
Browse files Browse the repository at this point in the history
fix: wrong name of ragnarok_common package
  • Loading branch information
mhordynski committed Sep 4, 2024
1 parent 2409b3c commit 0f347e0
Show file tree
Hide file tree
Showing 33 changed files with 25 additions and 25 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# RAG-Stack (to be renamed)
# Ragnarok

Repository for internal experiment with our upcoming LLM framework.

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,8 @@

from pydantic import BaseModel

from ragstack_common.llms.litellm import LiteLLM
from ragstack_common.prompt import Prompt
from ragnarok_common.llms.litellm import LiteLLM
from ragnarok_common.prompt import Prompt


class LoremPromptInput(BaseModel):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
from pydantic import BaseModel

from ragstack_common.prompt import Prompt
from ragnarok_common.prompt import Prompt


class LoremPromptInput(BaseModel):
Expand Down
File renamed without changes.
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
[metadata]
name = ragstack-common
name = ragnarok-common
version = 0.0.1
description = "common utilities for the ragstack framework"
description = "common utilities for the ragnarok framework"
author = deepsense.ai
author_email = [email protected]
license = Other/Proprietary License
Expand Down
File renamed without changes.
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
from functools import cached_property
from typing import Generic, Optional, Type, cast, overload

from ragstack_common.prompt.base import BasePrompt, BasePromptWithParser, OutputT
from ragnarok_common.prompt.base import BasePrompt, BasePromptWithParser, OutputT

from .clients.base import LLMClient, LLMClientOptions, LLMOptions

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@

from pydantic import BaseModel

from ragstack_common.prompt import ChatFormat
from ragnarok_common.prompt import ChatFormat

from ..types import NotGiven

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
HAS_LITELLM = False


from ragstack_common.prompt import ChatFormat
from ragnarok_common.prompt import ChatFormat

from ..types import NOT_GIVEN, NotGiven
from .base import LLMClient, LLMOptions
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
except ImportError:
HAS_LOCAL_LLM = False

from ragstack_common.prompt import ChatFormat
from ragnarok_common.prompt import ChatFormat

from ..types import NOT_GIVEN, NotGiven
from .base import LLMClient, LLMOptions
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
except ImportError:
HAS_LITELLM = False

from ragstack_common.prompt.base import BasePrompt
from ragnarok_common.prompt.base import BasePrompt

from .base import LLM
from .clients.litellm import LiteLLMClient, LiteLLMOptions
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
except ImportError:
HAS_LOCAL_LLM = False

from ragstack_common.prompt.base import BasePrompt
from ragnarok_common.prompt.base import BasePrompt

from .base import LLM
from .clients.local import LocalLLMClient, LocalLLMOptions
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
from ragnarok_common.prompt.prompt import ChatFormat, Prompt

__all__ = ["Prompt", "ChatFormat"]
File renamed without changes.
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
from pydantic import BaseModel

from ragstack_common.llms.clients.litellm import LiteLLMOptions
from ragstack_common.llms.litellm import LiteLLM
from ragstack_common.prompt import Prompt
from ragstack_common.prompt.base import BasePrompt, BasePromptWithParser, ChatFormat
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


class MockPrompt(BasePrompt):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,8 @@

import pytest

from ragstack_common.prompt import Prompt
from ragstack_common.prompt.parsers import ResponseParsingError
from ragnarok_common.prompt import Prompt
from ragnarok_common.prompt.parsers import ResponseParsingError

from .test_prompt import _PromptOutput

Expand Down
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import pydantic
import pytest

from ragstack_common.prompt import Prompt
from ragnarok_common.prompt import Prompt


class _PromptInput(pydantic.BaseModel):
Expand Down

This file was deleted.

4 changes: 2 additions & 2 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
multi_line_output = 3
line_length = 120
include_trailing_comma = true
known_first_party = ['ragstack_common']
known_first_party = ['ragnarok_common']
known_third_party = [ # Most popular libraries. Extend if necessary.
'IPython',
'PIL',
Expand Down Expand Up @@ -85,7 +85,7 @@ no_implicit_optional = true
mypy_path = ['packages']

[[tool.mypy.overrides]]
module = "ragstack_common.*"
module = "ragnarok_common.*"
ignore_missing_imports = false
disallow_untyped_defs = true

Expand Down
2 changes: 1 addition & 1 deletion requirements-dev.txt
Original file line number Diff line number Diff line change
Expand Up @@ -9,4 +9,4 @@ pytest_asyncio>=0.24.0
pip-licenses>=4.0.0,<5.0.0

# extras required by tests:
-e packages/ragstack-common[litellm]
-e packages/ragnarok-common[litellm]

0 comments on commit 0f347e0

Please sign in to comment.