mirror of
https://github.com/tiennm99/DocsGPT.git
synced 2026-10-15 08:16:38 +00:00
refactor: rename the application package to docsgpt
The backend import package is now docsgpt, the name it will carry on PyPI; application was far too generic to install into anyone's site-packages. git mv plus a mechanical rewrite of every import, dotted string and path reference: 734 Python files, the compose files, Dockerfile, workflows, docs, setup scripts, devcontainer, k8s manifests, vscode config, pytest and coverage config, .gitignore. Behaviour is unchanged. Kept for one release: - A top-level application package whose meta-path finder resolves application.x.y to the already-imported docsgpt.x.y object, so old imports and entry points (celery -A application.app.celery, uvicorn application.asgi:asgi_app) keep working with a FutureWarning. - Celery registers every application.* task name as an alias of its docsgpt.* task on start-up, so messages queued by the previous release still run. The redbeat key prefix moves to redbeat:docsgpt:v2: so schedule entries the previous release wrote are left unread instead of firing twice. The backend image builds from the repository root (docker build -f docsgpt/Dockerfile .) so it can ship the alias package; a root .dockerignore allow-lists docsgpt/ and application/ and keeps caches, local data, .env files, the sample index files and the Dockerfile out. Compose and the image workflows point at the new context.
This commit is contained in:
Alex committed
1 parent
3e7f1f91b4
commit
574f96341e
984 files changed
+9847
-9682
No files matched your search
@@ -2,7 +2,7 @@ from unittest.mock import MagicMock, patch
|
||||
|
||||
import pytest
|
||||
|
||||
from application.core.model_settings import (
|
||||
from docsgpt.core.model_settings import (
|
||||
AvailableModel,
|
||||
ModelCapabilities,
|
||||
ModelProvider,
|
||||
@@ -51,11 +51,11 @@ def _make_model(
|
||||
|
||||
class TestGetApiKeyForProvider:
|
||||
"""settings is lazily imported inside the function body, so we patch
|
||||
at application.core.settings.settings (the actual module attribute)."""
|
||||
at docsgpt.core.settings.settings (the actual module attribute)."""
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_openai_key(self):
|
||||
with patch("application.core.settings.settings") as mock_settings:
|
||||
with patch("docsgpt.core.settings.settings") as mock_settings:
|
||||
mock_settings.OPENAI_API_KEY = "sk-openai"
|
||||
mock_settings.API_KEY = "sk-fallback"
|
||||
mock_settings.OPEN_ROUTER_API_KEY = None
|
||||
@@ -65,94 +65,94 @@ class TestGetApiKeyForProvider:
|
||||
mock_settings.GROQ_API_KEY = None
|
||||
mock_settings.HUGGINGFACE_API_KEY = None
|
||||
|
||||
from application.core.model_utils import get_api_key_for_provider
|
||||
from docsgpt.core.model_utils import get_api_key_for_provider
|
||||
|
||||
assert get_api_key_for_provider("openai") == "sk-openai"
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_anthropic_key(self):
|
||||
with patch("application.core.settings.settings") as mock_settings:
|
||||
with patch("docsgpt.core.settings.settings") as mock_settings:
|
||||
mock_settings.ANTHROPIC_API_KEY = "sk-anthropic"
|
||||
mock_settings.API_KEY = "sk-fallback"
|
||||
|
||||
from application.core.model_utils import get_api_key_for_provider
|
||||
from docsgpt.core.model_utils import get_api_key_for_provider
|
||||
|
||||
assert get_api_key_for_provider("anthropic") == "sk-anthropic"
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_google_key(self):
|
||||
with patch("application.core.settings.settings") as mock_settings:
|
||||
with patch("docsgpt.core.settings.settings") as mock_settings:
|
||||
mock_settings.GOOGLE_API_KEY = "sk-google"
|
||||
mock_settings.API_KEY = "sk-fallback"
|
||||
|
||||
from application.core.model_utils import get_api_key_for_provider
|
||||
from docsgpt.core.model_utils import get_api_key_for_provider
|
||||
|
||||
assert get_api_key_for_provider("google") == "sk-google"
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_groq_key(self):
|
||||
with patch("application.core.settings.settings") as mock_settings:
|
||||
with patch("docsgpt.core.settings.settings") as mock_settings:
|
||||
mock_settings.GROQ_API_KEY = "sk-groq"
|
||||
mock_settings.API_KEY = "sk-fallback"
|
||||
|
||||
from application.core.model_utils import get_api_key_for_provider
|
||||
from docsgpt.core.model_utils import get_api_key_for_provider
|
||||
|
||||
assert get_api_key_for_provider("groq") == "sk-groq"
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_openrouter_key(self):
|
||||
with patch("application.core.settings.settings") as mock_settings:
|
||||
with patch("docsgpt.core.settings.settings") as mock_settings:
|
||||
mock_settings.OPEN_ROUTER_API_KEY = "sk-or"
|
||||
mock_settings.API_KEY = "sk-fallback"
|
||||
|
||||
from application.core.model_utils import get_api_key_for_provider
|
||||
from docsgpt.core.model_utils import get_api_key_for_provider
|
||||
|
||||
assert get_api_key_for_provider("openrouter") == "sk-or"
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_novita_key(self):
|
||||
with patch("application.core.settings.settings") as mock_settings:
|
||||
with patch("docsgpt.core.settings.settings") as mock_settings:
|
||||
mock_settings.NOVITA_API_KEY = "sk-novita"
|
||||
mock_settings.API_KEY = "sk-fallback"
|
||||
|
||||
from application.core.model_utils import get_api_key_for_provider
|
||||
from docsgpt.core.model_utils import get_api_key_for_provider
|
||||
|
||||
assert get_api_key_for_provider("novita") == "sk-novita"
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_huggingface_key(self):
|
||||
with patch("application.core.settings.settings") as mock_settings:
|
||||
with patch("docsgpt.core.settings.settings") as mock_settings:
|
||||
mock_settings.HUGGINGFACE_API_KEY = "hf-key"
|
||||
mock_settings.API_KEY = "sk-fallback"
|
||||
|
||||
from application.core.model_utils import get_api_key_for_provider
|
||||
from docsgpt.core.model_utils import get_api_key_for_provider
|
||||
|
||||
assert get_api_key_for_provider("huggingface") == "hf-key"
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_docsgpt_returns_fallback(self):
|
||||
with patch("application.core.settings.settings") as mock_settings:
|
||||
with patch("docsgpt.core.settings.settings") as mock_settings:
|
||||
mock_settings.API_KEY = "sk-fallback"
|
||||
|
||||
from application.core.model_utils import get_api_key_for_provider
|
||||
from docsgpt.core.model_utils import get_api_key_for_provider
|
||||
|
||||
assert get_api_key_for_provider("docsgpt") == "sk-fallback"
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_llama_cpp_returns_fallback(self):
|
||||
with patch("application.core.settings.settings") as mock_settings:
|
||||
with patch("docsgpt.core.settings.settings") as mock_settings:
|
||||
mock_settings.API_KEY = "sk-fallback"
|
||||
|
||||
from application.core.model_utils import get_api_key_for_provider
|
||||
from docsgpt.core.model_utils import get_api_key_for_provider
|
||||
|
||||
assert get_api_key_for_provider("llama.cpp") == "sk-fallback"
|
||||
|
||||
@pytest.mark.unit
|
||||
def test_unknown_provider_returns_fallback(self):
|
||||
with patch("application.core.settings.settings") as mock_settings:
|
||||
with patch("docsgpt.core.settings.settings") as mock_settings:
|
||||
mock_settings.API_KEY = "sk-fallback"
|
||||
|
||||
from application.core.model_utils import get_api_key_for_provider
|
||||
from docsgpt.core.model_utils import get_api_key_for_provider
|
||||
|
||||
assert get_api_key_for_provider("unknown_provider") == "sk-fallback"
|
||||
|
||||
@@ -161,7 +161,7 @@ class TestGetApiKeyForProvider:
|
||||
|
||||
class TestGetAllAvailableModels:
|
||||
@pytest.mark.unit
|
||||
@patch("application.core.model_utils.ModelRegistry.get_instance")
|
||||
@patch("docsgpt.core.model_utils.ModelRegistry.get_instance")
|
||||
def test_returns_enabled_models_as_dict(self, mock_get_instance):
|
||||
model_a = _make_model("model-a", display_name="Model A")
|
||||
model_b = _make_model("model-b", display_name="Model B")
|
||||
@@ -169,7 +169,7 @@ class TestGetAllAvailableModels:
|
||||
mock_registry.get_enabled_models.return_value = [model_a, model_b]
|
||||
mock_get_instance.return_value = mock_registry
|
||||
|
||||
from application.core.model_utils import get_all_available_models
|
||||
from docsgpt.core.model_utils import get_all_available_models
|
||||
|
||||
result = get_all_available_models()
|
||||
|
||||
@@ -179,13 +179,13 @@ class TestGetAllAvailableModels:
|
||||
assert result["model-b"]["display_name"] == "Model B"
|
||||
|
||||
@pytest.mark.unit
|
||||
@patch("application.core.model_utils.ModelRegistry.get_instance")
|
||||
@patch("docsgpt.core.model_utils.ModelRegistry.get_instance")
|
||||
def test_empty_registry(self, mock_get_instance):
|
||||
mock_registry = MagicMock()
|
||||
mock_registry.get_enabled_models.return_value = []
|
||||
mock_get_instance.return_value = mock_registry
|
||||
|
||||
from application.core.model_utils import get_all_available_models
|
||||
from docsgpt.core.model_utils import get_all_available_models
|
||||
|
||||
assert get_all_available_models() == {}
|
||||
|
||||
@@ -195,24 +195,24 @@ class TestGetAllAvailableModels:
|
||||
|
||||
class TestValidateModelId:
|
||||
@pytest.mark.unit
|
||||
@patch("application.core.model_utils.ModelRegistry.get_instance")
|
||||
@patch("docsgpt.core.model_utils.ModelRegistry.get_instance")
|
||||
def test_exists(self, mock_get_instance):
|
||||
mock_registry = MagicMock()
|
||||
mock_registry.model_exists.return_value = True
|
||||
mock_get_instance.return_value = mock_registry
|
||||
|
||||
from application.core.model_utils import validate_model_id
|
||||
from docsgpt.core.model_utils import validate_model_id
|
||||
|
||||
assert validate_model_id("gpt-4") is True
|
||||
|
||||
@pytest.mark.unit
|
||||
@patch("application.core.model_utils.ModelRegistry.get_instance")
|
||||
@patch("docsgpt.core.model_utils.ModelRegistry.get_instance")
|
||||
def test_not_exists(self, mock_get_instance):
|
||||
mock_registry = MagicMock()
|
||||
mock_registry.model_exists.return_value = False
|
||||
mock_get_instance.return_value = mock_registry
|
||||
|
||||
from application.core.model_utils import validate_model_id
|
||||
from docsgpt.core.model_utils import validate_model_id
|
||||
|
||||
assert validate_model_id("nonexistent") is False
|
||||
|
||||
@@ -222,7 +222,7 @@ class TestValidateModelId:
|
||||
|
||||
class TestGetModelCapabilities:
|
||||
@pytest.mark.unit
|
||||
@patch("application.core.model_utils.ModelRegistry.get_instance")
|
||||
@patch("docsgpt.core.model_utils.ModelRegistry.get_instance")
|
||||
def test_model_found(self, mock_get_instance):
|
||||
model = _make_model(
|
||||
"gpt-4",
|
||||
@@ -235,7 +235,7 @@ class TestGetModelCapabilities:
|
||||
mock_registry.get_model.return_value = model
|
||||
mock_get_instance.return_value = mock_registry
|
||||
|
||||
from application.core.model_utils import get_model_capabilities
|
||||
from docsgpt.core.model_utils import get_model_capabilities
|
||||
|
||||
caps = get_model_capabilities("gpt-4")
|
||||
|
||||
@@ -246,13 +246,13 @@ class TestGetModelCapabilities:
|
||||
assert caps["context_window"] == 8192
|
||||
|
||||
@pytest.mark.unit
|
||||
@patch("application.core.model_utils.ModelRegistry.get_instance")
|
||||
@patch("docsgpt.core.model_utils.ModelRegistry.get_instance")
|
||||
def test_model_not_found(self, mock_get_instance):
|
||||
mock_registry = MagicMock()
|
||||
mock_registry.get_model.return_value = None
|
||||
mock_get_instance.return_value = mock_registry
|
||||
|
||||
from application.core.model_utils import get_model_capabilities
|
||||
from docsgpt.core.model_utils import get_model_capabilities
|
||||
|
||||
assert get_model_capabilities("nonexistent") is None
|
||||
|
||||
@@ -262,13 +262,13 @@ class TestGetModelCapabilities:
|
||||
|
||||
class TestGetDefaultModelId:
|
||||
@pytest.mark.unit
|
||||
@patch("application.core.model_utils.ModelRegistry.get_instance")
|
||||
@patch("docsgpt.core.model_utils.ModelRegistry.get_instance")
|
||||
def test_returns_default(self, mock_get_instance):
|
||||
mock_registry = MagicMock()
|
||||
mock_registry.default_model_id = "gpt-4"
|
||||
mock_get_instance.return_value = mock_registry
|
||||
|
||||
from application.core.model_utils import get_default_model_id
|
||||
from docsgpt.core.model_utils import get_default_model_id
|
||||
|
||||
assert get_default_model_id() == "gpt-4"
|
||||
|
||||
@@ -278,25 +278,25 @@ class TestGetDefaultModelId:
|
||||
|
||||
class TestGetProviderFromModelId:
|
||||
@pytest.mark.unit
|
||||
@patch("application.core.model_utils.ModelRegistry.get_instance")
|
||||
@patch("docsgpt.core.model_utils.ModelRegistry.get_instance")
|
||||
def test_model_found(self, mock_get_instance):
|
||||
model = _make_model("gpt-4", provider=ModelProvider.OPENAI)
|
||||
mock_registry = MagicMock()
|
||||
mock_registry.get_model.return_value = model
|
||||
mock_get_instance.return_value = mock_registry
|
||||
|
||||
from application.core.model_utils import get_provider_from_model_id
|
||||
from docsgpt.core.model_utils import get_provider_from_model_id
|
||||
|
||||
assert get_provider_from_model_id("gpt-4") == "openai"
|
||||
|
||||
@pytest.mark.unit
|
||||
@patch("application.core.model_utils.ModelRegistry.get_instance")
|
||||
@patch("docsgpt.core.model_utils.ModelRegistry.get_instance")
|
||||
def test_model_not_found(self, mock_get_instance):
|
||||
mock_registry = MagicMock()
|
||||
mock_registry.get_model.return_value = None
|
||||
mock_get_instance.return_value = mock_registry
|
||||
|
||||
from application.core.model_utils import get_provider_from_model_id
|
||||
from docsgpt.core.model_utils import get_provider_from_model_id
|
||||
|
||||
assert get_provider_from_model_id("nonexistent") is None
|
||||
|
||||
@@ -306,28 +306,28 @@ class TestGetProviderFromModelId:
|
||||
|
||||
class TestGetTokenLimit:
|
||||
@pytest.mark.unit
|
||||
@patch("application.core.model_utils.ModelRegistry.get_instance")
|
||||
@patch("docsgpt.core.model_utils.ModelRegistry.get_instance")
|
||||
def test_model_found(self, mock_get_instance):
|
||||
model = _make_model("gpt-4", context_window=8192)
|
||||
mock_registry = MagicMock()
|
||||
mock_registry.get_model.return_value = model
|
||||
mock_get_instance.return_value = mock_registry
|
||||
|
||||
from application.core.model_utils import get_token_limit
|
||||
from docsgpt.core.model_utils import get_token_limit
|
||||
|
||||
assert get_token_limit("gpt-4") == 8192
|
||||
|
||||
@pytest.mark.unit
|
||||
@patch("application.core.model_utils.ModelRegistry.get_instance")
|
||||
@patch("docsgpt.core.model_utils.ModelRegistry.get_instance")
|
||||
def test_model_not_found_returns_default(self, mock_get_instance):
|
||||
mock_registry = MagicMock()
|
||||
mock_registry.get_model.return_value = None
|
||||
mock_get_instance.return_value = mock_registry
|
||||
|
||||
with patch("application.core.settings.settings") as mock_settings:
|
||||
with patch("docsgpt.core.settings.settings") as mock_settings:
|
||||
mock_settings.DEFAULT_LLM_TOKEN_LIMIT = 128000
|
||||
|
||||
from application.core.model_utils import get_token_limit
|
||||
from docsgpt.core.model_utils import get_token_limit
|
||||
|
||||
assert get_token_limit("nonexistent") == 128000
|
||||
|
||||
@@ -337,36 +337,36 @@ class TestGetTokenLimit:
|
||||
|
||||
class TestGetBaseUrlForModel:
|
||||
@pytest.mark.unit
|
||||
@patch("application.core.model_utils.ModelRegistry.get_instance")
|
||||
@patch("docsgpt.core.model_utils.ModelRegistry.get_instance")
|
||||
def test_model_with_base_url(self, mock_get_instance):
|
||||
model = _make_model("custom-model", base_url="http://localhost:8080")
|
||||
mock_registry = MagicMock()
|
||||
mock_registry.get_model.return_value = model
|
||||
mock_get_instance.return_value = mock_registry
|
||||
|
||||
from application.core.model_utils import get_base_url_for_model
|
||||
from docsgpt.core.model_utils import get_base_url_for_model
|
||||
|
||||
assert get_base_url_for_model("custom-model") == "http://localhost:8080"
|
||||
|
||||
@pytest.mark.unit
|
||||
@patch("application.core.model_utils.ModelRegistry.get_instance")
|
||||
@patch("docsgpt.core.model_utils.ModelRegistry.get_instance")
|
||||
def test_model_without_base_url(self, mock_get_instance):
|
||||
model = _make_model("gpt-4", base_url=None)
|
||||
mock_registry = MagicMock()
|
||||
mock_registry.get_model.return_value = model
|
||||
mock_get_instance.return_value = mock_registry
|
||||
|
||||
from application.core.model_utils import get_base_url_for_model
|
||||
from docsgpt.core.model_utils import get_base_url_for_model
|
||||
|
||||
assert get_base_url_for_model("gpt-4") is None
|
||||
|
||||
@pytest.mark.unit
|
||||
@patch("application.core.model_utils.ModelRegistry.get_instance")
|
||||
@patch("docsgpt.core.model_utils.ModelRegistry.get_instance")
|
||||
def test_model_not_found(self, mock_get_instance):
|
||||
mock_registry = MagicMock()
|
||||
mock_registry.get_model.return_value = None
|
||||
mock_get_instance.return_value = mock_registry
|
||||
|
||||
from application.core.model_utils import get_base_url_for_model
|
||||
from docsgpt.core.model_utils import get_base_url_for_model
|
||||
|
||||
assert get_base_url_for_model("nonexistent") is None
|
||||
Reference in new issue
Block a user