feat: Phase 1+2 — project setup and core infrastructure
- requirements.txt, config.yaml, .env, Dockerfile, docker-compose.yml
- app/core: config (YAML+env override), logging (JSON structured),
exceptions (typed hierarchy), json_utils (Markdown fence stripping)
- app/clients: LLMClient ABC + ZhipuAIClient (run_in_executor),
StorageClient ABC + RustFSClient (boto3 head_object for size check)
- app/main.py: FastAPI app with health endpoint and router registration
- app/core/dependencies.py: lru_cache singleton factories
- tests/conftest.py: mock_llm, mock_storage, test_app, client fixtures
- pytest.ini: asyncio_mode=auto
- 11 unit tests passing
2026-04-10 15:22:45 +08:00
|
|
|
from abc import ABC, abstractmethod
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class LLMClient(ABC):
|
|
|
|
|
@abstractmethod
|
|
|
|
|
async def chat(self, model: str, messages: list[dict]) -> str:
|
|
|
|
|
"""Send a text chat request and return the response content string."""
|
|
|
|
|
|
|
|
|
|
@abstractmethod
|
|
|
|
|
async def chat_vision(self, model: str, messages: list[dict]) -> str:
|
|
|
|
|
"""Send a multimodal (vision) chat request and return the response content string."""
|
2026-04-10 16:43:28 +08:00
|
|
|
|
|
|
|
|
@abstractmethod
|
|
|
|
|
async def submit_finetune(self, jsonl_url: str, base_model: str, hyperparams: dict) -> str:
|
|
|
|
|
"""Submit a fine-tune job and return the job_id."""
|
|
|
|
|
|
|
|
|
|
@abstractmethod
|
|
|
|
|
async def get_finetune_status(self, job_id: str) -> dict:
|
|
|
|
|
"""Return a dict with keys: job_id, status (raw SDK string), progress (int|None), error_message (str|None)."""
|