fixed the RAg in test pipeline issue

This commit is contained in:
belviskhoremk
2026-04-26 18:51:48 +00:00
parent 205d9d7901
commit 97a501097d
14 changed files with 249 additions and 57 deletions

View File

@@ -62,6 +62,7 @@ class UserResponse(BaseModel):
plan: str = "free"
is_admin: bool = False
created_at: Optional[datetime] = None
language: Optional[str] = "fr"
class TokenResponse(BaseModel):
@@ -101,7 +102,7 @@ class ChatbotCreate(BaseModel):
name: str = Field(min_length=2, max_length=100)
description: Optional[str] = None
system_prompt: Optional[str] = None
model: str = "accounts/fireworks/models/kimi-k2-instruct-0905"
model: str = "accounts/fireworks/models/kimi-k2-instruct"
@field_validator("name", mode="before")
@classmethod
@@ -301,6 +302,7 @@ class ChatResponse(BaseModel):
tokens_used: int = 0
needs_lead_capture: bool = False
handoff: bool = False
low_confidence: bool = False
class MessageResponse(BaseModel):
@@ -460,6 +462,24 @@ class FeedbackCreate(BaseModel):
feedback: str # 'positive' or 'negative'
# ─── Test Models ──────────────────────────────────────────────────────────────
class TestQuestion(BaseModel):
question: str
class TestChatRequest(BaseModel):
questions: List[str] = Field(min_length=1, max_length=10)
class TestChatResult(BaseModel):
question: str
response: str
confidence_score: float
sources: List[SourceDocument]
model_used: str
# ─── Inbox Models ─────────────────────────────────────────────────────────────
class InboxConversation(BaseModel):