lm-deluge 0.0.4__py3-none-any.whl → 0.0.5__py3-none-any.whl

lm_deluge/api_requests/anthropic.py

@@ -119,9 +119,7 @@ class AnthropicRequest(APIRequestBase):
         if status_code >= 200 and status_code < 300:
             try:
                 data = await http_response.json()
-                print("response data:", data)
                 content = data["content"]  # [0]["text"]
-                print("content is length", len(content))
                 for item in content:
                     if item["type"] == "text":
                         completion = item["text"]

lm_deluge/api_requests/cohere.py

@@ -55,7 +55,7 @@ class CohereRequest(APIRequestBase):
 
         self.model = APIModel.from_registry(model_name)
         self.url = f"{self.model.api_base}/chat"
-        self.system_message, chat_history, last_user_message = prompt.to_cohere()
+        messages = prompt.to_cohere()
 
         self.request_header = {
             "Authorization": f"bearer {os.getenv(self.model.api_key_env_var)}",
@@ -65,16 +65,12 @@ class CohereRequest(APIRequestBase):
 
         self.request_json = {
             "model": self.model.name,
-            "chat_history": chat_history,
-            "message": last_user_message,
+            "messages": messages,
             "temperature": sampling_params.temperature,
             "top_p": sampling_params.top_p,
             "max_tokens": sampling_params.max_new_tokens,
         }
 
-        if self.system_message:
-            self.request_json["preamble"] = self.system_message
-
     async def handle_response(self, http_response: ClientResponse) -> APIResponse:
         is_error = False
         error_message = None

{lm_deluge-0.0.4.dist-info → lm_deluge-0.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lm_deluge
-Version: 0.0.4
+Version: 0.0.5
 Summary: Python utility for using LLM API models.
 Author-email: Benjamin Anderson <ben@trytaylor.ai>
 Requires-Python: >=3.10

{lm_deluge-0.0.4.dist-info → lm_deluge-0.0.5.dist-info}/RECORD

@@ -12,9 +12,9 @@ lm_deluge/sampling_params.py,sha256=E2kewh1vz-1Qcy5xNBCzihfGgT_GcHYMfzaWb3FLiXs,
 lm_deluge/tool.py,sha256=RVUW3E3FW11jCM-R7pIL1GpRs1YKCOjvTkL1D5xPetk,3196
 lm_deluge/tracker.py,sha256=Dk99scN_NeDEO0gkLO5efXiZq11Ga-k6cerUHWN7IWY,1292
 lm_deluge/api_requests/__init__.py,sha256=_aSpD6CJL9g6OpLPoChXiHjl4MH_OlGcKgfZaW8cgLM,71
-lm_deluge/api_requests/anthropic.py,sha256=HHmDOHs9euIwbjUpN8lLIX1FyAR0-fJ3-t_vW8T_Y2Y,6721
+lm_deluge/api_requests/anthropic.py,sha256=URbiD-ANn_P3StFJVP2JoDWuoloZVsAUly8CGSyV2Kw,6618
 lm_deluge/api_requests/base.py,sha256=PGI6gMnT0ZDhlkk9YWBRTUiwqPLan-O7yYt0MKyFjy0,14753
-lm_deluge/api_requests/cohere.py,sha256=1bCnV3Ot_n7CJKYLU_AoaKUuTgPUXUSBHPkAauQBimY,5089
+lm_deluge/api_requests/cohere.py,sha256=KyYpbOjJxaTdauQBG_D7WWPLjkqmmk9-YFFnxYyT0vU,4898
 lm_deluge/api_requests/common.py,sha256=ZtUirAlYNE4CJOQPbNip-mWKOMXCJi2_malwYsUcqsg,642
 lm_deluge/api_requests/google.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lm_deluge/api_requests/openai.py,sha256=vjf2WFqQimroyGt_hX730lF8T83Bv29MzSyA8Ek0A9s,5663
@@ -31,7 +31,7 @@ lm_deluge/util/logprobs.py,sha256=UkBZakOxWluaLqHrjARu7xnJ0uCHVfLGHJdnYlEcutk,11
 lm_deluge/util/pdf.py,sha256=zBcuh2IJxRfro6JPzQkAqdc6hUcrduFwb9aEoacNG9U,1590
 lm_deluge/util/validation.py,sha256=hz5dDb3ebvZrZhnaWxOxbNSVMI6nmaOODBkk0htAUhs,1575
 lm_deluge/util/xml.py,sha256=-yUKOZwsAY009w4ppQ8FI4hb5gDvLzMwcSgv88aEnaE,10578
-lm_deluge-0.0.4.dist-info/METADATA,sha256=eHZFvQYWovoOURJX3BhW_PfV8W4cMLf6YRKn_Qxiwo4,4957
-lm_deluge-0.0.4.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-lm_deluge-0.0.4.dist-info/top_level.txt,sha256=hqU-TJX93yBwpgkDtYcXyLr3t7TLSCCZ_reytJjwBaE,10
-lm_deluge-0.0.4.dist-info/RECORD,,
+lm_deluge-0.0.5.dist-info/METADATA,sha256=MxI6OVGdYD6MsjeQPyoXDT8pneKB8FcZ652ZW0t_qyk,4957
+lm_deluge-0.0.5.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+lm_deluge-0.0.5.dist-info/top_level.txt,sha256=hqU-TJX93yBwpgkDtYcXyLr3t7TLSCCZ_reytJjwBaE,10
+lm_deluge-0.0.5.dist-info/RECORD,,