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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: llm-codegen-research
-Version: 2.10
+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.10 → 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 if temperature is not None else mistralai.UNSET,
+            temperature=temperature if temperature is not None else client.UNSET,
             top_p=top_p,
-            max_tokens=max_tokens if max_tokens is not None else 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.10 → 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.10
+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