chore: layout optimize

2026-03-29 14:44:32 +08:00
parent 74fb360df6
commit 22e1891a57
10 changed files with 1316 additions and 925 deletions
@@ -0,0 +1,28 @@
+from typing import Optional
+from pydantic import BaseModel, Field
+
+class EmbeddingModelConfigBase(BaseModel):
+    name: str = Field(..., description="Display name for the model configuration")
+    provider: str = Field("openai", description="Provider type (e.g. openai)")
+    model: str = Field(..., description="Model name (e.g. text-embedding-3-small)")
+    api_base: Optional[str] = None
+    api_key: Optional[str] = None
+
+class EmbeddingModelConfigCreate(EmbeddingModelConfigBase):
+    pass
+
+class EmbeddingModelConfigUpdate(BaseModel):
+    name: Optional[str] = None
+    provider: Optional[str] = None
+    model: Optional[str] = None
+    api_base: Optional[str] = None
+    api_key: Optional[str] = None
+
+class EmbeddingModelConfig(EmbeddingModelConfigBase):
+    id: str
+
+class EmbeddingModelConnectionTestRequest(BaseModel):
+    provider: str = Field("openai")
+    model: str = Field(...)
+    api_base: Optional[str] = None
+    api_key: Optional[str] = None