Spaces:
Running
Running
File size: 1,140 Bytes
9002555 0743bb0 9002555 0743bb0 9002555 0743bb0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
from pydantic import BaseModel, Field
from typing import List, Optional, Dict, Any
from llama_index.core.llms import MessageRole
class MetadataRequest(BaseModel):
title: str
category : str
author: str
year: int
publisher: str
class DeleteById(BaseModel):
id : str
class UserPromptRequest(BaseModel):
session_id : str
prompt : str
streaming : bool
class BotResponse(BaseModel):
role : str = "assistant"
content : str
raw_references : List
references : List
metadata : List
scores : List
class BotResponseStreaming(BaseModel):
role : str = "assistant"
content : Optional[str] = None
completed_content : Optional[str] = None
reference : Optional[str] = None
metadata : Optional[Dict] = None
score : Optional[float] = None
class TestStreaming(BaseModel):
role : str = "assistant"
content : str
class ChatMessage(BaseModel):
"""Chat message."""
role: MessageRole = MessageRole.ASSISTANT
content: Optional[Any] = ""
metadata: List
def __str__(self) -> str:
return f"{self.role.value}: {self.content}" |