not-again-ai 0.5.0__tar.gz → 0.5.1__tar.gz

{not_again_ai-0.5.0 → not_again_ai-0.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: not-again-ai
-Version: 0.5.0
+Version: 0.5.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.
 Home-page: https://github.com/DaveCoDev/not-again-ai
 License: MIT
@@ -76,7 +76,7 @@ The package is split into subpackages, so you can install only the parts you nee
 The base package includes only functions that have minimal external dependencies and are useful in a variety of situations such as parallelization and filesystem operations.
 
 ## LLM (Large Language Model)
-[README](https://github.com/DaveCoDev/not-again-ai/blob/main/readmes/llm.md)
+[README](https://github.com/DaveCoDev/not-again-ai/blob/main/readmes/llm.md), [Example Notebooks](https://github.com/DaveCoDev/not-again-ai/blob/main/notebooks/llm/)
 
 Supports OpenAI chat completions and text embeddings. Includes functions for creating chat completion prompts, token management, and context management.
 

{not_again_ai-0.5.0 → not_again_ai-0.5.1}/README.md

@@ -42,7 +42,7 @@ The package is split into subpackages, so you can install only the parts you nee
 The base package includes only functions that have minimal external dependencies and are useful in a variety of situations such as parallelization and filesystem operations.
 
 ## LLM (Large Language Model)
-[README](https://github.com/DaveCoDev/not-again-ai/blob/main/readmes/llm.md)
+[README](https://github.com/DaveCoDev/not-again-ai/blob/main/readmes/llm.md), [Example Notebooks](https://github.com/DaveCoDev/not-again-ai/blob/main/notebooks/llm/)
 
 Supports OpenAI chat completions and text embeddings. Includes functions for creating chat completion prompts, token management, and context management.
 

{not_again_ai-0.5.0 → not_again_ai-0.5.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "not-again-ai"
-version = "0.5.0"
+version = "0.5.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 = ["DaveCoDev <dave.co.dev@gmail.com>"]
 license = "MIT"
@@ -41,6 +41,9 @@ llm = ["openai", "python-liquid", "tiktoken"]
 statistics = ["numpy", "scikit-learn", "scipy"]
 viz = ["numpy", "pandas", "seaborn"]
 
+[tool.poetry.dev-dependencies]
+ipykernel = "*"
+
 [tool.poetry.group.nox.dependencies]
 nox-poetry = "*"
 

{not_again_ai-0.5.0 → not_again_ai-0.5.1}/src/not_again_ai/llm/chat_completion.py

@@ -64,9 +64,8 @@ def chat_completion(
                 this will be a list of dictionaries containing the token, logprob, and bytes for each token in the message.
             'choices' (list[dict], optional): A list of chat completion choices if n > 1 where each dict contains the above fields.
             'completion_tokens' (int): The number of tokens used by the model to generate the completion.
-                NOTE: If n > 1 this is the sum of all completions and thus will be same value in each dict.
-            'prompt_tokens' (int): The number of tokens in the generated response.
-                NOTE: If n > 1 this is the sum of all completions and thus will be same value in each dict.
+                NOTE: If n > 1 this is the sum of all completions.
+            'prompt_tokens' (int): The number of tokens in the messages sent to the model.
             'system_fingerprint' (str, optional): If seed is set, a unique identifier for the model used to generate the response.
     """
     response_format = {"type": "json_object"} if json_mode else None

{not_again_ai-0.5.0 → not_again_ai-0.5.1}/src/not_again_ai/llm/chat_completion_vision.py

@@ -41,9 +41,8 @@ def chat_completion_vision(
             'message' (str | dict): The content of the generated assistant message.
             'choices' (list[dict], optional): A list of chat completion choices if n > 1 where each dict contains the above fields.
             'completion_tokens' (int): The number of tokens used by the model to generate the completion.
-                NOTE: If n > 1 this is the sum of all completions and thus will be same value in each dict.
-            'prompt_tokens' (int): The number of tokens in the generated response.
-                NOTE: If n > 1 this is the sum of all completions and thus will be same value in each dict.
+                NOTE: If n > 1 this is the sum of all completions.
+            'prompt_tokens' (int): The number of tokens in the messages sent to the model.
             'system_fingerprint' (str, optional): If seed is set, a unique identifier for the model used to generate the response.
     """
     kwargs.update(

{not_again_ai-0.5.0 → not_again_ai-0.5.1}/src/not_again_ai/llm/prompts.py

@@ -127,7 +127,7 @@ def create_image_url(image_path: Path) -> str:
 
 
 def chat_prompt_vision(messages_unformatted: list[dict[str, Any]], variables: dict[str, str]) -> list[dict[str, Any]]:
-    """Formats a list of messages for OpenAI's chat completion API for vision models only using Liquid templating.
+    """Formats a list of messages for OpenAI's chat completion API, for vision models only, using Liquid templating.
 
     Args:
         messages_unformatted (list[dict[str, list[dict[str, Path | str]] | str]]):