not-again-ai 0.16.0__tar.gz → 0.16.1__tar.gz

{not_again_ai-0.16.0 → not_again_ai-0.16.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: not-again-ai
-Version: 0.16.0
+Version: 0.16.1
 Summary: Designed to once and for all collect all the little things that come up over and over again in AI projects and put them in one place.
 License: MIT
 Author: DaveCoDev
@@ -27,7 +27,7 @@ Requires-Dist: numpy (>=2.2) ; extra == "viz"
 Requires-Dist: ollama (>=0.4) ; extra == "llm"
 Requires-Dist: openai (>=1) ; extra == "llm"
 Requires-Dist: pandas (>=2.2) ; extra == "viz"
-Requires-Dist: playwright (>=1.49) ; extra == "data"
+Requires-Dist: playwright (>=1.50) ; extra == "data"
 Requires-Dist: pydantic (>=2.10)
 Requires-Dist: pytest-playwright (>=0.7) ; extra == "data"
 Requires-Dist: python-liquid (>=1.12) ; extra == "llm"

{not_again_ai-0.16.0 → not_again_ai-0.16.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "not-again-ai"
-version = "0.16.0"
+version = "0.16.1"
 description = "Designed to once and for all collect all the little things that come up over and over again in AI projects and put them in one place."
 authors = [
     { name = "DaveCoDev", email = "dave.co.dev@gmail.com" }
@@ -40,7 +40,7 @@ poetry-plugin-export = ">=1.8"
 
 [project.optional-dependencies]
 data = [
-    "playwright>=1.49",
+    "playwright>=1.50",
     "pytest-playwright>=0.7"
 ]
 llm = [

{not_again_ai-0.16.0 → not_again_ai-0.16.1}/src/not_again_ai/llm/chat_completion/providers/ollama_api.py

@@ -28,6 +28,7 @@ OLLAMA_PARAMETER_MAP = {
     "logit_bias": None,
     "top_logprobs": None,
     "presence_penalty": None,
+    "max_tokens": "num_predict",
 }
 
 
@@ -45,6 +46,10 @@ def validate(request: ChatCompletionRequest) -> None:
         logger.warning("Parameter 'stop' needs to be a string and not a list. It will be ignored.")
         request.stop = None
 
+    # Raise an error if both "max_tokens" and "max_completion_tokens" are provided
+    if request.max_tokens is not None and request.max_completion_tokens is not None:
+        raise ValueError("`max_tokens` and `max_completion_tokens` cannot both be provided.")
+
 
 def ollama_chat_completion(
     request: ChatCompletionRequest,

{not_again_ai-0.16.0 → not_again_ai-0.16.1}/src/not_again_ai/llm/chat_completion/providers/openai_api.py

@@ -31,6 +31,10 @@ def validate(request: ChatCompletionRequest) -> None:
     if request.json_mode and request.structured_outputs is not None:
         raise ValueError("json_schema and json_mode cannot be used together.")
 
+    # Raise an error if both "max_tokens" and "max_completion_tokens" are provided
+    if request.max_tokens is not None and request.max_completion_tokens is not None:
+        raise ValueError("`max_tokens` and `max_completion_tokens` cannot both be provided.")
+
 
 def openai_chat_completion(
     request: ChatCompletionRequest,

{not_again_ai-0.16.0 → not_again_ai-0.16.1}/src/not_again_ai/llm/chat_completion/types.py

@@ -118,6 +118,11 @@ class ChatCompletionRequest(BaseModel):
     top_k: int | None = Field(default=None)
     min_p: float | None = Field(default=None)
 
+    max_tokens: int | None = Field(
+        default=None,
+        description="Sometimes `max_completion_tokens` is not correctly supported so we provide this as a fallback.",
+    )
+
 
 class ChatCompletionChoice(BaseModel):
     message: AssistantMessage