Skip to content

Commit a6fb046

Browse files
fix missing default arg in ollama model contextsize to align with optional typing for use in contextsize calls in agent
1 parent cc0e2c0 commit a6fb046

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

archytas/models/ollama.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def ainvoke(self, input, *, config=None, stop=None, **kwargs):
3434
return super().ainvoke(input, config=config, stop=stop, **kwargs)
3535

3636
@lru_cache()
37-
def contextsize(self, model_name: str | None) -> int | None:
37+
def contextsize(self, model_name: str | None = None) -> int | None:
3838
if model_name is None:
3939
model_name = self.model_name
4040
show_response = self.model._client.show(self.model_name)

0 commit comments

Comments
 (0)