first commit

2026-03-06 21:11:10 +08:00
commit 927b8a6cac
144 changed files with 26301 additions and 0 deletions
--- a/llms/configs.py
+++ b/llms/configs.py
@@ -0,0 +1,34 @@
+from typing import Optional
+
+from pydantic import BaseModel, Field, field_validator
+
+
+class LlmConfig(BaseModel):
+    provider: str = Field(description="Provider of the LLM (e.g., 'ollama', 'openai')", default="openai")
+    config: Optional[dict] = Field(description="Configuration for the specific LLM", default={})
+
+    @field_validator("config")
+    def validate_config(cls, v, values):
+        provider = values.data.get("provider")
+        if provider in (
+            "openai",
+            "ollama",
+            "anthropic",
+            "groq",
+            "together",
+            "aws_bedrock",
+            "litellm",
+            "azure_openai",
+            "openai_structured",
+            "azure_openai_structured",
+            "gemini",
+            "deepseek",
+            "xai",
+            "sarvam",
+            "lmstudio",
+            "vllm",
+            "langchain",
+        ):
+            return v
+        else:
+            raise ValueError(f"Unsupported LLM provider: {provider}")