llama-index

1
from typing import Optional
2
from llama_index.core.base.response.schema import Response
3
from llama_index.core.bridge.pydantic import BaseModel
4
from pydantic import BaseModel as V2BaseModel
5

6

7
class ContributorQueryRequest(V2BaseModel):
8
    query: str
9

10

11
class ContributorQueryResponse(BaseModel):
12
    response: Optional[str]
13
    score: Optional[float]
14

15
    def __str__(self) -> str:
16
        """Convert to string representation."""
17
        return self.response or "None"
18

19
    def to_response(self) -> Response:
20
        """Convert to Response type."""
21
        return Response(response=self.response, metadata={"score": self.score})
22