mirror of
http://88.130.71.182:3000/BlitTech/contexta_be.git
synced 2026-06-12 23:23:21 +00:00
fixed the RAg in test pipeline issue
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user