eliott/ollama-python
1
0
Fork 0
mirror of https://github.com/kharonsec/ollama-python synced 2026-04-25 15:14:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

Pydantic Fixes and Tests (#311)

Browse Source 
* Added SubscriptableBaseModel to the Model classes and added Image codec test

---------

Co-authored-by: Parth Sareen <parth@Parths-MacBook-Pro.local>
This commit is contained in:
Parth Sareen
2024-11-06 14:04:56 -08:00
committed by Michael Yang
parent 0bbc246007
commit f25834217b
2 changed files with 18 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
ollama/_types.py
View File

@@ -97,6 +97,7 @@ class BaseGenerateRequest(BaseStreamableRequest):
class Image(BaseModel):
value: Union[FilePath, Base64Str, bytes]
# This overloads the `model_dump` method and returns values depending on the type of the `value` field
@model_serializer
def serialize_model(self):
if isinstance(self.value, Path):
@@ -333,7 +334,7 @@ class ModelDetails(SubscriptableBaseModel):
class ListResponse(SubscriptableBaseModel):
class Model(BaseModel):
class Model(SubscriptableBaseModel):
modified_at: Optional[datetime] = None
digest: Optional[str] = None
size: Optional[ByteSize] = None
@@ -394,7 +395,7 @@ class ShowResponse(SubscriptableBaseModel):
class ProcessResponse(SubscriptableBaseModel):
class Model(BaseModel):
class Model(SubscriptableBaseModel):
model: Optional[str] = None
name: Optional[str] = None
digest: Optional[str] = None