fix: update codebase for Pydantic v2 compatibility (BaseSettings, Config, validators)

app/core/config.py

@@ -3,9 +3,9 @@ Configuration management for Sema Chat API
 Environment-driven settings for flexible model backends
 """
 
-import os
 from typing import List, Optional
-from pydantic import BaseSettings, Field
+from pydantic import Field
+from pydantic_settings import BaseSettings
 from dotenv import load_dotenv
 
 # Load environment variables from .env file
@@ -132,9 +132,10 @@ class Settings(BaseSettings):
     system_prompt_code: Optional[str] = Field(default=None, env="SYSTEM_PROMPT_CODE")
     system_prompt_creative: Optional[str] = Field(default=None, env="SYSTEM_PROMPT_CREATIVE")
 
-    class Config:
-        env_file = ".env"
-        case_sensitive = False
+    model_config = {
+        "env_file": ".env",
+        "case_sensitive": False
+    }
 
     def get_system_prompt(self, prompt_type: str = "default") -> str:
         """Get system prompt based on type"""

app/models/schemas.py

@@ -2,21 +2,21 @@
 Pydantic models for request/response validation
 """
 
-from typing import List, Optional, Dict, Any, Union
-from pydantic import BaseModel, Field, validator
+from typing import List, Optional, Dict, Any
+from pydantic import BaseModel, Field, field_validator, ConfigDict
 from datetime import datetime
-import uuid
 
 
 class ChatMessage(BaseModel):
     """Individual chat message model"""
-    
+
     role: str = Field(..., description="Message role: 'user' or 'assistant'")
     content: str = Field(..., description="Message content")
     timestamp: datetime = Field(default_factory=datetime.utcnow, description="Message timestamp")
     metadata: Optional[Dict[str, Any]] = Field(default=None, description="Additional message metadata")
-    
-    @validator('role')
+
+    @field_validator('role')
+    @classmethod
     def validate_role(cls, v):
         if v not in ['user', 'assistant', 'system']:
             raise ValueError('Role must be user, assistant, or system')
@@ -25,7 +25,7 @@ class ChatMessage(BaseModel):
 
 class ChatRequest(BaseModel):
     """Chat request model"""
-    
+
     message: str = Field(
         ...,
         description="User message",
@@ -63,7 +63,7 @@ class ChatRequest(BaseModel):
 
 class ChatResponse(BaseModel):
     """Chat response model"""
-    
+
     message: str = Field(..., description="Assistant response message")
     session_id: str = Field(..., description="Session identifier")
     message_id: str = Field(..., description="Unique message identifier")
@@ -72,9 +72,9 @@ class ChatResponse(BaseModel):
     generation_time: float = Field(..., description="Time taken to generate response (seconds)")
     token_count: Optional[int] = Field(default=None, description="Number of tokens in response")
     finish_reason: Optional[str] = Field(default=None, description="Reason generation finished")
-    
-    class Config:
-        json_schema_extra = {
+
+    model_config = ConfigDict(
+        json_schema_extra={
             "example": {
                 "message": "Hello! I'm doing well, thank you for asking. How can I help you today?",
                 "session_id": "user-123-session",
@@ -86,11 +86,12 @@ class ChatResponse(BaseModel):
                 "finish_reason": "stop"
             }
         }
+    )
 
 
 class StreamChunk(BaseModel):
     """Streaming response chunk model"""
-    
+
     content: str = Field(..., description="Chunk content")
     session_id: str = Field(..., description="Session identifier")
     message_id: str = Field(..., description="Message identifier")
@@ -101,15 +102,15 @@ class StreamChunk(BaseModel):
 
 class ConversationHistory(BaseModel):
     """Conversation history model"""
-    
+
     session_id: str = Field(..., description="Session identifier")
     messages: List[ChatMessage] = Field(..., description="List of messages in conversation")
     created_at: datetime = Field(default_factory=datetime.utcnow, description="Session creation time")
     updated_at: datetime = Field(default_factory=datetime.utcnow, description="Last update time")
     message_count: int = Field(..., description="Total number of messages")
-    
-    class Config:
-        json_schema_extra = {
+
+    model_config = ConfigDict(
+        json_schema_extra={
             "example": {
                 "session_id": "user-123-session",
                 "messages": [
@@ -129,11 +130,12 @@ class ConversationHistory(BaseModel):
                 "message_count": 2
             }
         }
+    )
 
 
 class SessionInfo(BaseModel):
     """Session information model"""
-    
+
     session_id: str = Field(..., description="Session identifier")
     created_at: datetime = Field(..., description="Session creation time")
     updated_at: datetime = Field(..., description="Last activity time")
@@ -144,7 +146,7 @@ class SessionInfo(BaseModel):
 
 class HealthResponse(BaseModel):
     """Health check response model"""
-    
+
     status: str = Field(..., description="API health status")
     version: str = Field(..., description="API version")
     model_type: str = Field(..., description="Current model backend type")
@@ -153,9 +155,9 @@ class HealthResponse(BaseModel):
     uptime: float = Field(..., description="API uptime in seconds")
     active_sessions: int = Field(..., description="Number of active chat sessions")
     timestamp: datetime = Field(default_factory=datetime.utcnow, description="Health check timestamp")
-    
-    class Config:
-        json_schema_extra = {
+
+    model_config = ConfigDict(
+        json_schema_extra={
             "example": {
                 "status": "healthy",
                 "version": "1.0.0",
@@ -167,19 +169,20 @@ class HealthResponse(BaseModel):
                 "timestamp": "2024-01-15T10:30:00Z"
             }
         }
+    )
 
 
 class ErrorResponse(BaseModel):
     """Error response model"""
-    
+
     error: str = Field(..., description="Error type")
     message: str = Field(..., description="Error message")
     details: Optional[Dict[str, Any]] = Field(default=None, description="Additional error details")
     timestamp: datetime = Field(default_factory=datetime.utcnow, description="Error timestamp")
     request_id: Optional[str] = Field(default=None, description="Request identifier for debugging")
-    
-    class Config:
-        json_schema_extra = {
+
+    model_config = ConfigDict(
+        json_schema_extra={
             "example": {
                 "error": "validation_error",
                 "message": "Message content is required",
@@ -188,19 +191,20 @@ class ErrorResponse(BaseModel):
                 "request_id": "req-123-456"
             }
         }
+    )
 
 
 class ModelInfo(BaseModel):
     """Model information model"""
-    
+
     name: str = Field(..., description="Model name")
     type: str = Field(..., description="Model backend type")
     loaded: bool = Field(..., description="Whether model is loaded")
     parameters: Optional[Dict[str, Any]] = Field(default=None, description="Model parameters")
     capabilities: List[str] = Field(default=[], description="Model capabilities")
-    
-    class Config:
-        json_schema_extra = {
+
+    model_config = ConfigDict(
+        json_schema_extra={
             "example": {
                 "name": "TinyLlama/TinyLlama-1.1B-Chat-v1.0",
                 "type": "local",
@@ -213,3 +217,4 @@ class ModelInfo(BaseModel):
                 "capabilities": ["chat", "instruction_following", "streaming"]
             }
         }
+    )

fix_deployment.md

@@ -0,0 +1,131 @@
+# 🔧 Quick Fix for HuggingFace Deployment
+
+## ❌ **Issue Identified**
+The build failed due to **Pydantic v2 compatibility issues**:
+- `BaseSettings` moved to `pydantic-settings` package
+- `Config` class syntax changed to `model_config`
+- `@validator` changed to `@field_validator`
+
+## ✅ **Fixes Applied**
+
+### 1. Updated requirements.txt
+```diff
++ pydantic-settings
+```
+
+### 2. Fixed app/core/config.py
+```diff
+- from pydantic import BaseSettings, Field
++ from pydantic import Field
++ from pydantic_settings import BaseSettings
+
+- class Config:
+-     env_file = ".env"
+-     case_sensitive = False
++ model_config = {
++     "env_file": ".env", 
++     "case_sensitive": False
++ }
+```
+
+### 3. Fixed app/models/schemas.py
+```diff
+- from pydantic import BaseModel, Field, validator
++ from pydantic import BaseModel, Field, field_validator, ConfigDict
+
+- @validator('role')
+- def validate_role(cls, v):
++ @field_validator('role')
++ @classmethod
++ def validate_role(cls, v):
+
+- class Config:
+-     json_schema_extra = {...}
++ model_config = ConfigDict(
++     json_schema_extra={...}
++ )
+```
+
+## 🚀 **Quick Deployment Fix**
+
+### Option 1: Update Your Existing Space
+1. Go to your HuggingFace Space: https://huggingface.co/spaces/sematech/sema-chat
+2. Click "Files" tab
+3. Update these files with the fixed versions:
+   - `requirements.txt`
+   - `app/core/config.py` 
+   - `app/models/schemas.py`
+
+### Option 2: Re-run Setup Script
+```bash
+cd backend/sema-chat
+./setup_huggingface.sh
+```
+Then push the updated files to your Space.
+
+### Option 3: Manual Git Update
+```bash
+# Clone your space
+git clone https://huggingface.co/spaces/sematech/sema-chat
+cd sema-chat
+
+# Copy fixed files
+cp /path/to/sema/backend/sema-chat/requirements.txt .
+cp /path/to/sema/backend/sema-chat/app/core/config.py app/core/
+cp /path/to/sema/backend/sema-chat/app/models/schemas.py app/models/
+
+# Commit and push
+git add .
+git commit -m "Fix Pydantic v2 compatibility issues
+
+- Add pydantic-settings dependency
+- Update BaseSettings import
+- Fix Config class syntax
+- Update validator decorators"
+git push
+```
+
+## 🎯 **Environment Variables for Gemma**
+
+Make sure these are set in your Space settings:
+```
+MODEL_TYPE=local
+MODEL_NAME=google/gemma-2b-it
+DEVICE=cpu
+TEMPERATURE=0.7
+MAX_NEW_TOKENS=256
+DEBUG=false
+ENVIRONMENT=production
+```
+
+**Alternative (API-based):**
+```
+MODEL_TYPE=google
+MODEL_NAME=gemma-2-9b-it
+GOOGLE_API_KEY=your_google_api_key_here
+DEBUG=false
+ENVIRONMENT=production
+```
+
+## 🧪 **Test After Fix**
+
+Once the Space rebuilds successfully:
+```bash
+# Health check
+curl https://sematech-sema-chat.hf.space/health
+
+# Chat test
+curl -X POST "https://sematech-sema-chat.hf.space/api/v1/chat" \
+  -H "Content-Type: application/json" \
+  -d '{"message": "Hello! Can you introduce yourself?", "session_id": "test"}'
+```
+
+## 📋 **Build Status**
+- ❌ **Before**: Pydantic import errors
+- ✅ **After**: Should build successfully with Pydantic v2
+
+The fixes ensure compatibility with the latest Pydantic version while maintaining all functionality.
+
+---
+
+**Your Space should now deploy successfully! 🚀**

requirements.txt

@@ -1,6 +1,7 @@
 fastapi
 uvicorn
 pydantic
+pydantic-settings
 python-multipart
 websockets
 sse-starlette
@@ -23,6 +24,5 @@ openai
 anthropic
 
 # Utilities
-uuid
 asyncio-mqtt
 redis