mirror of
https://github.com/aimingmed/aimingmed-ai.git
synced 2026-01-31 11:49:45 +08:00
32 lines
963 B
Python
32 lines
963 B
Python
from typing import Literal
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class RouteQuery(BaseModel):
|
|
"""Route a user query to the most relevant datasource."""
|
|
|
|
datasource: Literal["vectorstore", "web_search"] = Field(
|
|
...,
|
|
description="Given a user question choose to route it to web search or a vectorstore.",
|
|
)
|
|
|
|
class GradeDocuments(BaseModel):
|
|
"""Binary score for relevance check on retrieved documents."""
|
|
|
|
binary_score: str = Field(
|
|
description="Documents are relevant to the question, 'yes' or 'no'"
|
|
)
|
|
|
|
class GradeHallucinations(BaseModel):
|
|
"""Binary score for hallucination present in generation answer."""
|
|
|
|
binary_score: str = Field(
|
|
description="Answer is grounded in the facts, 'yes' or 'no'"
|
|
)
|
|
|
|
class GradeAnswer(BaseModel):
|
|
"""Binary score to assess answer addresses question."""
|
|
|
|
binary_score: str = Field(
|
|
description="Answer addresses the question, 'yes' or 'no'"
|
|
) |