mirror of
https://github.com/james-m-jordan/morphik-core.git
synced 2025-05-09 19:32:38 +00:00
20 lines
447 B
Python
20 lines
447 B
Python
from pydantic import BaseModel
|
|
from typing import Dict, List, Optional
|
|
|
|
|
|
class CompletionResponse(BaseModel):
|
|
"""Response from completion generation"""
|
|
|
|
completion: str
|
|
usage: Dict[str, int]
|
|
finish_reason: Optional[str] = None
|
|
|
|
|
|
class CompletionRequest(BaseModel):
|
|
"""Request for completion generation"""
|
|
|
|
query: str
|
|
context_chunks: List[str]
|
|
max_tokens: Optional[int] = 1000
|
|
temperature: Optional[float] = 0.7
|