deploy

schemas/schema.py

@@ -1,5 +1,5 @@
-from pydantic import BaseModel
-from typing import List, Optional
+from pydantic import BaseModel, Field, ConfigDict
+from typing import List
 
 class SMSRequest(BaseModel):
     messages: List[str]
@@ -8,14 +8,14 @@ class EmbeddingResponse(BaseModel):
     dimensions: int
     embeddings: List[List[float]]
 
-    class Config:
-        
-        json_schema_extra = {
+    model_config = ConfigDict(
+        json_schema_extra={
             "example": {
                 "dimensions": 768,
                 "embeddings": [
-                    [0.1, 0.2, 0.3, ...],  # 768 values for each message
-                    [0.4, 0.5, 0.6, ...]
+                    [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, ...],  # 768 values for each message
+                    [0.7, 0.8, 0.9, 1.0, 1.1, 1.2, ...]
                 ]
             }
         }
+    )