Add many parameters to API endpoints

Support conversational HuggingFace providers
Fix streaming in PollinationsAI provider
This commit is contained in:
hlohaus
2025-03-11 22:16:03 +01:00
parent 3e7af90949
commit 713ad2c83c
7 changed files with 211 additions and 158 deletions

View File

@@ -18,6 +18,9 @@ class ChatCompletionsConfig(BaseModel):
image_name: Optional[str] = None
images: Optional[list[tuple[str, str]]] = None
temperature: Optional[float] = None
presence_penalty: Optional[float] = None
frequency_penalty: Optional[float] = None
top_p: Optional[float] = None
max_tokens: Optional[int] = None
stop: Union[list[str], str, None] = None
api_key: Optional[str] = None
@@ -27,7 +30,6 @@ class ChatCompletionsConfig(BaseModel):
conversation_id: Optional[str] = None
conversation: Optional[dict] = None
history_disabled: Optional[bool] = None
auto_continue: Optional[bool] = None
timeout: Optional[int] = None
tool_calls: list = Field(default=[], examples=[[
{
@@ -48,6 +50,14 @@ class ImageGenerationConfig(BaseModel):
response_format: Optional[str] = None
api_key: Optional[str] = None
proxy: Optional[str] = None
width: Optional[int] = None
height: Optional[int] = None
num_inference_steps: Optional[int] = None
seed: Optional[int] = None
guidance_scale: Optional[int] = None
aspect_ratio: Optional[str] = None
n: Optional[int] = None
negative_prompt: Optional[str] = None
class ProviderResponseModel(BaseModel):
id: str