UPDATE

rag_chat.py

@@ -0,0 +1,50 @@
+from __future__ import annotations
+
+from typing import Literal
+from uuid import UUID
+
+from pydantic import Field
+
+from .common import ApiEnvelope, JsonObject, JsonValue, RouteContract, SchemaModel
+
+HTTP_METHOD = 'POST'
+ROUTE_PATH = '/api/rag/chat/'
+
+
+class RagChatRequest(SchemaModel):
+    farm_uuid: UUID
+    query: str | None = None
+    message: str | None = None
+    history: list[JsonObject] | str | None = None
+    image_urls: list[str] = Field(default_factory=list)
+    image: str | None = None
+    images: list[str] = Field(default_factory=list)
+
+
+class RagChatSection(SchemaModel):
+    type: Literal['recommendation', 'list', 'warning', 'info', 'summary']
+    title: str
+    icon: str | None = None
+    content: str | None = None
+    items: list[str] = Field(default_factory=list)
+    primaryAction: str | None = None
+    timing: str | None = None
+    validityPeriod: str | None = None
+    expandableExplanation: str | None = None
+    metadata: JsonValue | None = None
+
+
+class RagChatResponseData(SchemaModel):
+    sections: list[RagChatSection] = Field(default_factory=list)
+
+
+class RagChatResponse(ApiEnvelope[RagChatResponseData]):
+    pass
+
+
+CONTRACT = RouteContract(
+    method=HTTP_METHOD,
+    path=ROUTE_PATH,
+    request_model=RagChatRequest.__name__,
+    response_model=RagChatResponse.__name__,
+)