llm-codegen-research 2.9.1__tar.gz → 2.11__tar.gz

{llm_codegen_research-2.9.1 → llm_codegen_research-2.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: llm-codegen-research
-Version: 2.9.1
+Version: 2.11
 Summary: Useful classes and methods for researching code-generation by LLMs.
 Author-email: Lukas Twist <itsluketwist@gmail.com>
 Project-URL: Homepage, https://github.com/itsluketwist/llm-codegen-research

{llm_codegen_research-2.9.1 → llm_codegen_research-2.11}/src/llm_cgr/llm/clients/anthropic.py

@@ -71,8 +71,8 @@ class Anthropic_LLM(Base_LLM):
             model=model,
             system=system or self._system or anthropic.NOT_GIVEN,
             messages=input,
-            temperature=temperature or anthropic.NOT_GIVEN,
-            top_p=top_p or anthropic.NOT_GIVEN,
-            max_tokens=max_tokens or DEFAULT_MAX_TOKENS,
+            temperature=temperature if temperature is not None else anthropic.NOT_GIVEN,
+            top_p=top_p if top_p is not None else anthropic.NOT_GIVEN,
+            max_tokens=max_tokens if max_tokens is not None else DEFAULT_MAX_TOKENS,
         )
         return response.content[0].text

{llm_codegen_research-2.9.1 → llm_codegen_research-2.11}/src/llm_cgr/llm/clients/deepseek.py

@@ -69,8 +69,8 @@ class DeepSeek_LLM(Base_LLM):
         response = self._client.chat.completions.create(
             messages=input,
             model=model,
-            temperature=temperature or openai.omit,
-            top_p=top_p or openai.omit,
-            max_completion_tokens=max_tokens or openai.omit,
+            temperature=temperature if temperature is not None else openai.omit,
+            top_p=top_p if top_p is not None else openai.omit,
+            max_completion_tokens=max_tokens if max_tokens is not None else openai.omit,
         )
         return response.choices[0].message.content

{llm_codegen_research-2.9.1 → llm_codegen_research-2.11}/src/llm_cgr/llm/clients/mistral.py

@@ -3,7 +3,7 @@
 import os
 from typing import Any
 
-import mistralai
+from mistralai import client
 
 from llm_cgr.llm.clients.base import Base_LLM
 
@@ -31,7 +31,7 @@ class Mistral_LLM(Base_LLM):
             top_p=top_p,
             max_tokens=max_tokens,
         )
-        self._client = mistralai.Mistral(
+        self._client = client.Mistral(
             api_key=os.environ["MISTRAL_API_KEY"],
         )
 
@@ -71,8 +71,8 @@ class Mistral_LLM(Base_LLM):
         response = self._client.chat.complete(
             model=model,
             messages=input,
-            temperature=temperature or mistralai.UNSET,
+            temperature=temperature if temperature is not None else client.UNSET,
             top_p=top_p,
-            max_tokens=max_tokens or mistralai.UNSET,
+            max_tokens=max_tokens if max_tokens is not None else client.UNSET,
         )
         return response.choices[0].message.content

{llm_codegen_research-2.9.1 → llm_codegen_research-2.11}/src/llm_cgr/llm/clients/nscale.py

@@ -69,8 +69,8 @@ class Nscale_LLM(Base_LLM):
         response = self._client.chat.completions.create(
             messages=input,
             model=model,
-            temperature=temperature or openai.omit,
-            top_p=top_p or openai.omit,
-            max_completion_tokens=max_tokens or openai.omit,
+            temperature=temperature if temperature is not None else openai.omit,
+            top_p=top_p if top_p is not None else openai.omit,
+            max_completion_tokens=max_tokens if max_tokens is not None else openai.omit,
         )
         return response.choices[0].message.content

{llm_codegen_research-2.9.1 → llm_codegen_research-2.11}/src/llm_cgr/llm/clients/openai.py

@@ -67,8 +67,8 @@ class OpenAI_LLM(Base_LLM):
         response = self._client.responses.create(
             input=cast(list[ResponseInputItemParam], input),
             model=model,
-            temperature=temperature or openai.omit,
-            top_p=top_p or openai.omit,
-            max_output_tokens=max_tokens or openai.omit,
+            temperature=temperature if temperature is not None else openai.omit,
+            top_p=top_p if top_p is not None else openai.omit,
+            max_output_tokens=max_tokens if max_tokens is not None else openai.omit,
         )
         return response.output_text

{llm_codegen_research-2.9.1 → llm_codegen_research-2.11}/src/llm_codegen_research.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: llm-codegen-research
-Version: 2.9.1
+Version: 2.11
 Summary: Useful classes and methods for researching code-generation by LLMs.
 Author-email: Lukas Twist <itsluketwist@gmail.com>
 Project-URL: Homepage, https://github.com/itsluketwist/llm-codegen-research