cat-llm 0.0.67__py3-none-any.whl → 0.0.69__py3-none-any.whl

{cat_llm-0.0.67.dist-info → cat_llm-0.0.69.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cat-llm
-Version: 0.0.67
+Version: 0.0.69
 Summary: A tool for categorizing text data and images using LLMs and vision models
 Project-URL: Documentation, https://github.com/chrissoria/cat-llm#readme
 Project-URL: Issues, https://github.com/chrissoria/cat-llm/issues
@@ -29,7 +29,7 @@ Description-Content-Type: text/markdown
 
 ![catllm Logo](https://github.com/chrissoria/cat-llm/blob/main/images/logo.png?raw=True)
 
-# catllm
+# cat-llm
 
 [![PyPI - Version](https://img.shields.io/pypi/v/cat-llm.svg)](https://pypi.org/project/cat-llm)
 [![PyPI - Python Version](https://img.shields.io/pypi/pyversions/cat-llm.svg)](https://pypi.org/project/cat-llm)

{cat_llm-0.0.67.dist-info → cat_llm-0.0.69.dist-info}/RECORD

@@ -1,15 +1,19 @@
 catllm/CERAD_functions.py,sha256=q4HbP5e2Yu8NnZZ-2eX4sImyj6u3i8xWcq0pYU81iis,22676
-catllm/__about__.py,sha256=hIVt-fQydvImRffcCyiHlTFeOR5F2aEchTWXAsWTPc4,430
+catllm/__about__.py,sha256=qQkN04YWoxAJ5HglANO-XGwexy9aL_qFoZSv_CueaUs,430
 catllm/__init__.py,sha256=sf02zp7N0NW0mAQi7eQ4gliWR1EwoqvXkHN2HwwjcTE,372
 catllm/build_web_research.py,sha256=880dfE2bEQb-FrXP-42JoLLtyc9ox_sBULDr38xiTiQ,22655
 catllm/image_functions.py,sha256=8_FftRU285x1HT-AgNkaobefQVD-5q7ZY_t7JFdL3Sg,36177
-catllm/text_functions.py,sha256=_GMretLVBUs0ntL-wV6My0TRAvzvaPo_WU9ZqXlKxeM,19426
+catllm/model_reference_list.py,sha256=37pWwMcgnf4biE3BVRluH5oz2P6ccdJJiCVNHodBH8k,2307
+catllm/text_functions.py,sha256=Vd9tAPDCDEhoXVW6O-jXeftJiZQmsyyrKeEUneYeobw,32533
+catllm/calls/CoVe.py,sha256=Y9OGJbaeJ3Odwira92cPXUlnm_ADFqvpOSFSNjFzMMU,10847
+catllm/calls/__init__.py,sha256=fWuMwLeSGa6zXJYd4s8IyNblsD62G-1NMUsOKrNIkoI,725
+catllm/calls/all_calls.py,sha256=E25KpZ_MakMDeCpNCOOM8kQvlfex6UMjnGN1wHkA4AI,14356
 catllm/images/circle.png,sha256=JWujAWAh08-TajAoEr_TAeFNLlfbryOLw6cgIBREBuQ,86202
 catllm/images/cube.png,sha256=nFec3e5bmRe4zrBCJ8QK-HcJLrG7u7dYdKhmdMfacfE,77275
 catllm/images/diamond.png,sha256=rJDZKtsnBGRO8FPA0iHuA8FvHFGi9PkI_DWSFdw6iv0,99568
 catllm/images/overlapping_pentagons.png,sha256=VO5plI6eoVRnjfqinn1nNzsCP2WQhuQy71V0EASouW4,71208
 catllm/images/rectangles.png,sha256=2XM16HO9EYWj2yHgN4bPXaCwPfl7iYQy0tQUGaJX9xg,40692
-cat_llm-0.0.67.dist-info/METADATA,sha256=UR95eJdArTmJe6A2g2hH0Q9mZ9PkEQsJ2kVSg3KfQe4,22423
-cat_llm-0.0.67.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-cat_llm-0.0.67.dist-info/licenses/LICENSE,sha256=Vje2sS5WV4TnIwY5uQHrF4qnBAM3YOk1pGpdH0ot-2o,34969
-cat_llm-0.0.67.dist-info/RECORD,,
+cat_llm-0.0.69.dist-info/METADATA,sha256=E2q6apmvq1sDDiisnfyyQZzxqjNnqjCSecpalb5MgWQ,22424
+cat_llm-0.0.69.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+cat_llm-0.0.69.dist-info/licenses/LICENSE,sha256=Vje2sS5WV4TnIwY5uQHrF4qnBAM3YOk1pGpdH0ot-2o,34969
+cat_llm-0.0.69.dist-info/RECORD,,

catllm/__about__.py

@@ -1,7 +1,7 @@
 # SPDX-FileCopyrightText: 2025-present Christopher Soria <chrissoria@berkeley.edu>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "0.0.67"
+__version__ = "0.0.69"
 __author__ = "Chris Soria"
 __email__ = "chrissoria@berkeley.edu"
 __title__ = "cat-llm"

catllm/calls/CoVe.py

@@ -0,0 +1,304 @@
+# openai chain of verification calls
+
+def chain_of_verification_openai(
+    initial_reply,
+    step2_prompt,
+    step3_prompt,
+    step4_prompt,
+    client,
+    user_model,
+    creativity,
+    remove_numbering
+):
+    """
+    Execute Chain of Verification (CoVe) process.
+    Returns the verified reply or initial reply if error occurs.
+    """
+    try:
+        # STEP 2: Generate verification questions
+        step2_filled = step2_prompt.replace('<<INITIAL_REPLY>>', initial_reply)
+        
+        verification_response = client.chat.completions.create(
+            model=user_model,
+            messages=[{'role': 'user', 'content': step2_filled}],
+            **({"temperature": creativity} if creativity is not None else {})
+        )
+        
+        verification_questions = verification_response.choices[0].message.content
+        
+        # STEP 3: Answer verification questions
+        questions_list = [
+            remove_numbering(q) 
+            for q in verification_questions.split('\n') 
+            if q.strip()
+        ]
+        verification_qa = []
+        
+        # Prompting each question individually
+        for question in questions_list:
+            step3_filled = step3_prompt.replace('<<QUESTION>>', question)
+            
+            answer_response = client.chat.completions.create(
+                model=user_model,
+                messages=[{'role': 'user', 'content': step3_filled}],
+                **({"temperature": creativity} if creativity is not None else {})
+            )
+            
+            answer = answer_response.choices[0].message.content
+            verification_qa.append(f"Q: {question}\nA: {answer}")
+        
+        # STEP 4: Final corrected categorization
+        verification_qa_text = "\n\n".join(verification_qa)
+        
+        step4_filled = (step4_prompt
+            .replace('<<INITIAL_REPLY>>', initial_reply)
+            .replace('<<VERIFICATION_QA>>', verification_qa_text))
+        
+        print(f"Final prompt:\n{step4_filled}\n")
+        
+        final_response = client.chat.completions.create(
+            model=user_model,
+            messages=[{'role': 'user', 'content': step4_filled}],
+            **({"temperature": creativity} if creativity is not None else {})
+        )
+        
+        verified_reply = final_response.choices[0].message.content
+        print("Chain of verification completed. Final response generated.\n")
+        
+        return verified_reply
+        
+    except Exception as e:
+        print(f"ERROR in Chain of Verification: {str(e)}")
+        print("Falling back to initial response.\n")
+        return initial_reply
+    
+# anthropic chain of verification calls
+
+def chain_of_verification_anthropic(
+    initial_reply,
+    step2_prompt,
+    step3_prompt,
+    step4_prompt,
+    client,
+    user_model,
+    creativity,
+    remove_numbering
+):
+    """
+    Execute Chain of Verification (CoVe) process for Anthropic Claude.
+    Returns the verified reply or initial reply if error occurs.
+    """
+    try:
+        # STEP 2: Generate verification questions
+        step2_filled = step2_prompt.replace('<<INITIAL_REPLY>>', initial_reply)
+        
+        verification_response = client.messages.create(
+            model=user_model,
+            messages=[{'role': 'user', 'content': step2_filled}],
+            max_tokens=4096,
+            **({"temperature": creativity} if creativity is not None else {})
+        )
+        
+        verification_questions = verification_response.content[0].text
+        
+        # STEP 3: Answer verification questions
+        questions_list = [
+            remove_numbering(q) 
+            for q in verification_questions.split('\n') 
+            if q.strip()
+        ]
+        print(f"Verification questions:\n{questions_list}\n")
+        verification_qa = []
+        
+        # Prompting each question individually
+        for question in questions_list:
+            step3_filled = step3_prompt.replace('<<QUESTION>>', question)
+            
+            answer_response = client.messages.create(
+                model=user_model,
+                messages=[{'role': 'user', 'content': step3_filled}],
+                max_tokens=4096,
+                **({"temperature": creativity} if creativity is not None else {})
+            )
+            
+            answer = answer_response.content[0].text
+            verification_qa.append(f"Q: {question}\nA: {answer}")
+        
+        # STEP 4: Final corrected categorization
+        verification_qa_text = "\n\n".join(verification_qa)
+        
+        step4_filled = (step4_prompt
+            .replace('<<INITIAL_REPLY>>', initial_reply)
+            .replace('<<VERIFICATION_QA>>', verification_qa_text))
+        
+        print(f"Final prompt:\n{step4_filled}\n")
+        
+        final_response = client.messages.create(
+            model=user_model,
+            messages=[{'role': 'user', 'content': step4_filled}],
+            max_tokens=4096,
+            **({"temperature": creativity} if creativity is not None else {})
+        )
+        
+        verified_reply = final_response.content[0].text
+        print("Chain of verification completed. Final response generated.\n")
+        
+        return verified_reply
+        
+    except Exception as e:
+        print(f"ERROR in Chain of Verification: {str(e)}")
+        print("Falling back to initial response.\n")
+        return initial_reply
+
+# google chain of verification calls
+def chain_of_verification_google(
+    initial_reply,
+    prompt,
+    step2_prompt,
+    step3_prompt,
+    step4_prompt,
+    url,
+    headers,
+    creativity,
+    remove_numbering,
+    make_google_request
+):
+    import time
+    """
+    Execute Chain of Verification (CoVe) process for Google Gemini.
+    Returns the verified reply or initial reply if error occurs.
+    """
+    try:
+        # STEP 2: Generate verification questions
+        step2_filled = step2_prompt.replace('<<INITIAL_REPLY>>', initial_reply)
+        
+        payload_step2 = {
+            "contents": [{
+                "parts": [{"text": step2_filled}]
+            }],
+            **({"generationConfig": {"temperature": creativity}} if creativity is not None else {})
+        }
+        
+        result_step2 = make_google_request(url, headers, payload_step2)
+        verification_questions = result_step2["candidates"][0]["content"]["parts"][0]["text"]
+        
+        # STEP 3: Answer verification questions
+        questions_list = [
+            remove_numbering(q) 
+            for q in verification_questions.split('\n') 
+            if q.strip()
+        ]
+        verification_qa = []
+        
+        for question in questions_list:
+            time.sleep(2)  # temporary rate limit handling
+            step3_filled = step3_prompt.replace('<<QUESTION>>', question)
+            
+            payload_step3 = {
+                "contents": [{
+                    "parts": [{"text": step3_filled}]
+                }],
+                **({"generationConfig": {"temperature": creativity}} if creativity is not None else {})
+            }
+            
+            result_step3 = make_google_request(url, headers, payload_step3)
+            answer = result_step3["candidates"][0]["content"]["parts"][0]["text"]
+            verification_qa.append(f"Q: {question}\nA: {answer}")
+        
+        # STEP 4: Final corrected categorization
+        verification_qa_text = "\n\n".join(verification_qa)
+        
+        step4_filled = (step4_prompt
+            .replace('<<PROMPT>>', prompt)
+            .replace('<<INITIAL_REPLY>>', initial_reply)
+            .replace('<<VERIFICATION_QA>>', verification_qa_text))
+        
+        payload_step4 = {
+            "contents": [{
+                "parts": [{"text": step4_filled}]
+            }],
+            **({"generationConfig": {"temperature": creativity}} if creativity is not None else {})
+        }
+        
+        result_step4 = make_google_request(url, headers, payload_step4)
+        verified_reply = result_step4["candidates"][0]["content"]["parts"][0]["text"]
+        
+        print("Chain of verification completed. Final response generated.\n")
+        return verified_reply
+        
+    except Exception as e:
+        print(f"ERROR in Chain of Verification: {str(e)}")
+        print("Falling back to initial response.\n")
+        return initial_reply
+    
+# mistral chain of verification calls
+
+def chain_of_verification_mistral(
+    initial_reply,
+    step2_prompt,
+    step3_prompt,
+    step4_prompt,
+    client,
+    user_model,
+    creativity,
+    remove_numbering
+):
+    """
+    Execute Chain of Verification (CoVe) process for Mistral AI.
+    Returns the verified reply or initial reply if error occurs.
+    """
+    try:
+        # STEP 2: Generate verification questions
+        step2_filled = step2_prompt.replace('<<INITIAL_REPLY>>', initial_reply)
+        
+        verification_response = client.chat.complete(
+            model=user_model,
+            messages=[{'role': 'user', 'content': step2_filled}],
+            **({"temperature": creativity} if creativity is not None else {})
+        )
+        
+        verification_questions = verification_response.choices[0].message.content
+        
+        # STEP 3: Answer verification questions
+        questions_list = [
+            remove_numbering(q) 
+            for q in verification_questions.split('\n') 
+            if q.strip()
+        ]
+        verification_qa = []
+        
+        # Prompting each question individually
+        for question in questions_list:
+            step3_filled = step3_prompt.replace('<<QUESTION>>', question)
+            
+            answer_response = client.chat.complete(
+                model=user_model,
+                messages=[{'role': 'user', 'content': step3_filled}],
+                **({"temperature": creativity} if creativity is not None else {})
+            )
+            
+            answer = answer_response.choices[0].message.content
+            verification_qa.append(f"Q: {question}\nA: {answer}")
+        
+        # STEP 4: Final corrected categorization
+        verification_qa_text = "\n\n".join(verification_qa)
+        
+        step4_filled = (step4_prompt
+            .replace('<<INITIAL_REPLY>>', initial_reply)
+            .replace('<<VERIFICATION_QA>>', verification_qa_text))
+        
+        final_response = client.chat.complete(
+            model=user_model,
+            messages=[{'role': 'user', 'content': step4_filled}],
+            **({"temperature": creativity} if creativity is not None else {})
+        )
+        
+        verified_reply = final_response.choices[0].message.content
+        print("Chain of verification completed. Final response generated.\n")
+        
+        return verified_reply
+        
+    except Exception as e:
+        print(f"ERROR in Chain of Verification: {str(e)}")
+        print("Falling back to initial response.\n")
+        return initial_reply

catllm/calls/__init__.py

@@ -0,0 +1,25 @@
+# SPDX-FileCopyrightText: 2025-present Christopher Soria <chrissoria@berkeley.edu>
+#
+# SPDX-License-Identifier: MIT
+
+from .all_calls import (
+    get_stepback_insight_openai,
+    get_stepback_insight_anthropic,
+    get_stepback_insight_google,
+    get_stepback_insight_mistral,
+    chain_of_verification_openai,
+    chain_of_verification_google,
+    chain_of_verification_anthropic,
+    chain_of_verification_mistral
+)
+
+__all__ = [
+    'get_stepback_insight_openai',
+    'get_stepback_insight_anthropic',
+    'get_stepback_insight_google',
+    'get_stepback_insight_mistral',
+    'chain_of_verification_openai',
+    'chain_of_verification_anthropic',
+    'chain_of_verification_google',
+    'chain_of_verification_mistral',
+]