mirror of
https://github.com/snailyp/gemini-balance.git
synced 2026-06-07 08:39:55 +08:00
21 lines
568 B
Python
21 lines
568 B
Python
from pydantic import BaseModel
|
|
from typing import List, Optional, Union
|
|
|
|
|
|
class ChatRequest(BaseModel):
|
|
messages: List[dict]
|
|
model: str = "gemini-1.5-flash-002"
|
|
temperature: Optional[float] = 0.7
|
|
stream: Optional[bool] = False
|
|
tools: Optional[List[dict]] = []
|
|
max_tokens: Optional[int] = 8192
|
|
stop: Optional[List[str]] = []
|
|
top_p: Optional[float] = 0.9
|
|
top_k: Optional[int] = 40
|
|
|
|
|
|
class EmbeddingRequest(BaseModel):
|
|
input: Union[str, List[str]]
|
|
model: str = "text-embedding-004"
|
|
encoding_format: Optional[str] = "float"
|