not-again-ai 0.12.0__py3-none-any.whl → 0.12.1__py3-none-any.whl

not_again_ai/local_llm/chat_completion.py

@@ -2,7 +2,7 @@ from typing import Any
 
 from azure.ai.inference import ChatCompletionsClient
 from ollama import Client
-from openai import OpenAI
+from openai import AzureOpenAI, OpenAI
 
 from not_again_ai.llm.gh_models import chat_completion as chat_completion_gh_models
 from not_again_ai.llm.openai_api import chat_completion as chat_completion_openai
@@ -12,7 +12,7 @@ from not_again_ai.local_llm.ollama import chat_completion as chat_completion_oll
 def chat_completion(
     messages: list[dict[str, Any]],
     model: str,
-    client: OpenAI | Client | ChatCompletionsClient,
+    client: OpenAI | AzureOpenAI | Client | ChatCompletionsClient,
     tools: list[dict[str, Any]] | None = None,
     max_tokens: int | None = None,
     temperature: float = 0.7,
@@ -27,7 +27,7 @@ def chat_completion(
     Args:
         messages (list[dict[str, Any]]): A list of messages to send to the model.
         model (str): The model name to use.
-        client (OpenAI | Client): The client object to use for chat completion.
+        client (OpenAI | AzureOpenAI | Client | ChatCompletionsClient): The client object to use for chat completion.
         tools (list[dict[str, Any]], optional):A list of tools the model may call.
             Use this to provide a list of functions the model may generate JSON inputs for. Defaults to None.
         max_tokens (int, optional): The maximum number of tokens to generate.
@@ -48,7 +48,7 @@ def chat_completion(
             extras (dict): This will contain any additional fields returned by corresponding provider.
     """
     # Determine which chat_completion function to call based on the client type
-    if isinstance(client, OpenAI):
+    if isinstance(client, OpenAI | AzureOpenAI):
         response = chat_completion_openai.chat_completion(
             messages=messages,
             model=model,

{not_again_ai-0.12.0.dist-info → not_again_ai-0.12.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: not-again-ai
-Version: 0.12.0
+Version: 0.12.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
@@ -27,7 +27,7 @@ Requires-Dist: jinja2 (>=3.1,<4.0) ; extra == "local-llm"
 Requires-Dist: loguru (==0.7.2)
 Requires-Dist: numpy (>=1.26,<2.0) ; extra == "statistics" or extra == "viz"
 Requires-Dist: ollama (>=0.3,<0.4) ; extra == "local-llm"
-Requires-Dist: openai (>=1.40,<2.0) ; extra == "llm"
+Requires-Dist: openai (>=1.41,<2.0) ; extra == "llm"
 Requires-Dist: pandas (>=2.2,<3.0) ; extra == "viz"
 Requires-Dist: pydantic (>=2.8,<3.0) ; extra == "llm"
 Requires-Dist: python-liquid (>=1.12,<2.0) ; extra == "llm"
@@ -35,7 +35,7 @@ Requires-Dist: scikit-learn (>=1.5,<2.0) ; extra == "statistics"
 Requires-Dist: scipy (>=1.14,<2.0) ; extra == "statistics"
 Requires-Dist: seaborn (>=0.13,<0.14) ; extra == "viz"
 Requires-Dist: tiktoken (>=0.7,<0.8) ; extra == "llm"
-Requires-Dist: transformers (>=4.43,<5.0) ; extra == "local-llm"
+Requires-Dist: transformers (>=4.44,<5.0) ; extra == "local-llm"
 Project-URL: Documentation, https://github.com/DaveCoDev/not-again-ai
 Project-URL: Repository, https://github.com/DaveCoDev/not-again-ai
 Description-Content-Type: text/markdown
@@ -77,6 +77,10 @@ The package is split into subpackages, so you can install only the parts you nee
     1. OpenAI API
         1. Go to https://platform.openai.com/settings/profile?tab=api-keys to get your API key.
         1. (Optional) Set the `OPENAI_API_KEY` and the `OPENAI_ORG_ID` environment variables.
+    1. Azure OpenAI (AOAI)
+       1. Using AOAI requires using Entra ID authentication. See https://learn.microsoft.com/en-us/azure/ai-services/openai/how-to/managed-identity for how to set this up for your AOAI deployment.
+          1. Requires the correct role assigned to your user account and being signed into the Azure CLI.
+       1. (Optional) Set the `AZURE_OPENAI_ENDPOINT` environment variable.
     1. GitHub Models
        1. Get a Personal Access Token from https://github.com/settings/tokens and set the `GITHUB_TOKEN` environment variable. The token does not need any permissions.
        1. Check the [Github Marketplace](https://github.com/marketplace/models) to see which models are available.

{not_again_ai-0.12.0.dist-info → not_again_ai-0.12.1.dist-info}/RECORD

@@ -14,7 +14,7 @@ not_again_ai/llm/openai_api/openai_client.py,sha256=AK9SDBkpP94u5Q73-Q5i5HRPQh_D
 not_again_ai/llm/openai_api/prompts.py,sha256=B62xs3WKaTv7SfT_TVC-PqO9oeWWpO0xS4_oxW9MYMQ,7093
 not_again_ai/llm/openai_api/tokens.py,sha256=RYBzl5vqE_MzWM60QbWC_6X9YOQoOgBOeR-68rM34II,4421
 not_again_ai/local_llm/__init__.py,sha256=BsUn39U3QQaw6yomQHfp_HIPHRIBoMAgjcP3CDADx04,882
-not_again_ai/local_llm/chat_completion.py,sha256=qcmRXF2cNpGO8X5SxNzXo7B9Z3pcEVCxuJ4lwkq7r-I,4749
+not_again_ai/local_llm/chat_completion.py,sha256=PmICXrGZJXIuqY00ULBGi2bKnPG8ticqTXZHSTzZK9o,4828
 not_again_ai/local_llm/huggingface/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 not_again_ai/local_llm/huggingface/chat_completion.py,sha256=Y6uMbxLG8TaMVi3hJGrMl_G9Y1N_0dld5Kv1iqYnoao,2300
 not_again_ai/local_llm/huggingface/helpers.py,sha256=YPr8KbQ8Ac_Mn_nBcrFuL3bCl-IuDCdaRvYVCocy8Gk,734
@@ -35,8 +35,8 @@ not_again_ai/viz/distributions.py,sha256=OyWwJaNI6lMRm_iSrhq-CORLNvXfeuLSgDtVo3u
 not_again_ai/viz/scatterplot.py,sha256=5CUOWeknbBOaZPeX9oPin5sBkRKEwk8qeFH45R-9LlY,2292
 not_again_ai/viz/time_series.py,sha256=pOGZqXp_2nd6nKo-PUQNCtmMh__69jxQ6bQibTGLwZA,5212
 not_again_ai/viz/utils.py,sha256=hN7gwxtBt3U6jQni2K8j5m5pCXpaJDoNzGhBBikEU28,238
-not_again_ai-0.12.0.dist-info/LICENSE,sha256=btjOgNGpp-ux5xOo1Gx1MddxeWtT9sof3s3Nui29QfA,1071
-not_again_ai-0.12.0.dist-info/METADATA,sha256=EOVISRpYFCS6UJyJWG9dL_Eb6N-CspGd9hPM6gHBUls,15984
-not_again_ai-0.12.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-not_again_ai-0.12.0.dist-info/entry_points.txt,sha256=EMJegugnmJUd-jMUA_qIRMIPAasbei8gP6O4-ER0BxQ,61
-not_again_ai-0.12.0.dist-info/RECORD,,
+not_again_ai-0.12.1.dist-info/LICENSE,sha256=btjOgNGpp-ux5xOo1Gx1MddxeWtT9sof3s3Nui29QfA,1071
+not_again_ai-0.12.1.dist-info/METADATA,sha256=VydzFufICQyP6paN15KJTudJi6rSpwWn5H_W1v46p6Y,16389
+not_again_ai-0.12.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+not_again_ai-0.12.1.dist-info/entry_points.txt,sha256=EMJegugnmJUd-jMUA_qIRMIPAasbei8gP6O4-ER0BxQ,61
+not_again_ai-0.12.1.dist-info/RECORD,,