style: 项目重构
1.项目改名为kilostar(千星) 2.后端部分进行大规模重构 3.node功能进行大规模重新设计
This commit is contained in:
@@ -27,10 +27,10 @@ def mock_import(name, globals=None, locals=None, fromlist=(), level=0):
|
||||
builtins.__import__ = mock_import
|
||||
|
||||
for mod in list(sys.modules.keys()):
|
||||
if "pretor.core.global_state_machine.global_state_machine" in mod or "ray" in mod:
|
||||
if "kilostar.core.global_state_machine.global_state_machine" in mod or "ray" in mod:
|
||||
del sys.modules[mod]
|
||||
|
||||
from pretor.core.global_state_machine.global_state_machine import GlobalStateMachine # noqa: E402
|
||||
from kilostar.core.global_state_machine.global_state_machine import GlobalStateMachine # noqa: E402
|
||||
|
||||
builtins.__import__ = real_import
|
||||
|
||||
@@ -74,7 +74,7 @@ async def test_add_provider_success(gsm, mock_postgres):
|
||||
@pytest.mark.asyncio
|
||||
async def test_add_provider_unsupported(gsm):
|
||||
gsm._global_provider_manager.provider_mapper = {}
|
||||
with patch("pretor.utils.logger.global_logger.bind") as mock_bind:
|
||||
with patch("kilostar.utils.logger.global_logger.bind") as mock_bind:
|
||||
mock_logger = MagicMock()
|
||||
mock_bind.return_value = mock_logger
|
||||
await gsm.add_provider_wrap("magic", "title", "url", "key", "1")
|
||||
@@ -91,8 +91,8 @@ async def test_add_provider_request_error(gsm):
|
||||
)
|
||||
gsm._global_provider_manager.provider_mapper = {"openai": mock_provider_class}
|
||||
|
||||
with patch("pretor.utils.logger.global_logger.bind") as mock_bind:
|
||||
from pretor.utils.error import RetryableError
|
||||
with patch("kilostar.utils.logger.global_logger.bind") as mock_bind:
|
||||
from kilostar.utils.error import RetryableError
|
||||
import pytest
|
||||
|
||||
mock_logger = MagicMock()
|
||||
@@ -108,7 +108,7 @@ async def test_add_provider_generic_error(gsm):
|
||||
mock_provider_class.create_provider.side_effect = ValueError("Some Error")
|
||||
gsm._global_provider_manager.provider_mapper = {"openai": mock_provider_class}
|
||||
|
||||
with patch("pretor.utils.logger.global_logger.bind") as mock_bind:
|
||||
with patch("kilostar.utils.logger.global_logger.bind") as mock_bind:
|
||||
mock_logger = MagicMock()
|
||||
mock_bind.return_value = mock_logger
|
||||
await gsm.add_provider_wrap("openai", "title", "url", "key", "1")
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
from pretor.core.global_state_machine.model_provider.base_provider import (
|
||||
from kilostar.core.global_state_machine.model_provider.base_provider import (
|
||||
Provider,
|
||||
ProviderArgs,
|
||||
ProviderStatus,
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
import pytest
|
||||
from unittest.mock import patch, MagicMock, AsyncMock
|
||||
from pretor.core.global_state_machine.model_provider.claude_provider import (
|
||||
from kilostar.core.global_state_machine.model_provider.claude_provider import (
|
||||
ClaudeProvider,
|
||||
ProviderArgs,
|
||||
)
|
||||
@@ -18,7 +18,7 @@ def provider_args():
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@patch(
|
||||
"pretor.core.global_state_machine.model_provider.claude_provider.httpx.AsyncClient"
|
||||
"kilostar.core.global_state_machine.model_provider.claude_provider.httpx.AsyncClient"
|
||||
)
|
||||
async def test_load_models_success(mock_client, provider_args):
|
||||
mock_response = MagicMock()
|
||||
@@ -37,7 +37,7 @@ async def test_load_models_success(mock_client, provider_args):
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@patch(
|
||||
"pretor.core.global_state_machine.model_provider.claude_provider.httpx.AsyncClient"
|
||||
"kilostar.core.global_state_machine.model_provider.claude_provider.httpx.AsyncClient"
|
||||
)
|
||||
async def test_load_models_error(mock_client, provider_args):
|
||||
mock_client_instance = AsyncMock()
|
||||
@@ -50,7 +50,7 @@ async def test_load_models_error(mock_client, provider_args):
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@patch(
|
||||
"pretor.core.global_state_machine.model_provider.claude_provider.ClaudeProvider._load_models",
|
||||
"kilostar.core.global_state_machine.model_provider.claude_provider.ClaudeProvider._load_models",
|
||||
return_value=["claude-3"],
|
||||
)
|
||||
async def test_create_provider(mock_load, provider_args):
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
import pytest
|
||||
from unittest.mock import patch, MagicMock, AsyncMock
|
||||
from pretor.core.global_state_machine.model_provider.openai_provider import (
|
||||
from kilostar.core.global_state_machine.model_provider.openai_provider import (
|
||||
OpenAIProvider,
|
||||
ProviderArgs,
|
||||
)
|
||||
@@ -28,7 +28,7 @@ def provider_args_no_v1():
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@patch(
|
||||
"pretor.core.global_state_machine.model_provider.openai_provider.httpx.AsyncClient"
|
||||
"kilostar.core.global_state_machine.model_provider.openai_provider.httpx.AsyncClient"
|
||||
)
|
||||
async def test_load_models_success(mock_client, provider_args):
|
||||
mock_response = MagicMock()
|
||||
@@ -51,7 +51,7 @@ async def test_load_models_success(mock_client, provider_args):
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@patch(
|
||||
"pretor.core.global_state_machine.model_provider.openai_provider.httpx.AsyncClient"
|
||||
"kilostar.core.global_state_machine.model_provider.openai_provider.httpx.AsyncClient"
|
||||
)
|
||||
async def test_load_models_no_v1(mock_client, provider_args_no_v1):
|
||||
mock_response = MagicMock()
|
||||
@@ -72,7 +72,7 @@ async def test_load_models_no_v1(mock_client, provider_args_no_v1):
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@patch(
|
||||
"pretor.core.global_state_machine.model_provider.openai_provider.httpx.AsyncClient"
|
||||
"kilostar.core.global_state_machine.model_provider.openai_provider.httpx.AsyncClient"
|
||||
)
|
||||
async def test_load_models_status_error(mock_client, provider_args):
|
||||
mock_response = MagicMock()
|
||||
@@ -88,7 +88,7 @@ async def test_load_models_status_error(mock_client, provider_args):
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@patch(
|
||||
"pretor.core.global_state_machine.model_provider.openai_provider.httpx.AsyncClient"
|
||||
"kilostar.core.global_state_machine.model_provider.openai_provider.httpx.AsyncClient"
|
||||
)
|
||||
async def test_load_models_request_error(mock_client, provider_args):
|
||||
import httpx
|
||||
@@ -100,7 +100,7 @@ async def test_load_models_request_error(mock_client, provider_args):
|
||||
mock_client.return_value.__aenter__.return_value = mock_client_instance
|
||||
|
||||
import pytest
|
||||
from pretor.utils.error import RetryableError
|
||||
from kilostar.utils.error import RetryableError
|
||||
|
||||
with pytest.raises(RetryableError):
|
||||
await OpenAIProvider._load_models(provider_args)
|
||||
@@ -108,7 +108,7 @@ async def test_load_models_request_error(mock_client, provider_args):
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@patch(
|
||||
"pretor.core.global_state_machine.model_provider.openai_provider.httpx.AsyncClient"
|
||||
"kilostar.core.global_state_machine.model_provider.openai_provider.httpx.AsyncClient"
|
||||
)
|
||||
async def test_load_models_generic_error(mock_client, provider_args):
|
||||
mock_client_instance = AsyncMock()
|
||||
@@ -121,7 +121,7 @@ async def test_load_models_generic_error(mock_client, provider_args):
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@patch(
|
||||
"pretor.core.global_state_machine.model_provider.openai_provider.OpenAIProvider._load_models",
|
||||
"kilostar.core.global_state_machine.model_provider.openai_provider.OpenAIProvider._load_models",
|
||||
return_value=["gpt-4"],
|
||||
)
|
||||
async def test_create_provider(mock_load, provider_args):
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
import pytest
|
||||
from unittest.mock import MagicMock, AsyncMock
|
||||
from pretor.core.global_state_machine.provider_manager import ProviderManager
|
||||
from kilostar.core.global_state_machine.provider_manager import ProviderManager
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
from pretor.core.global_state_machine.tool_manager import GlobalToolManager
|
||||
from kilostar.core.global_state_machine.tool_manager import GlobalToolManager
|
||||
|
||||
|
||||
def test_global_tool_manager_init():
|
||||
|
||||
Reference in New Issue
Block a user