cat-stack 0.1.0__py3-none-any.whl

cat_stack/calls/all_calls.py

@@ -0,0 +1,622 @@
+# openai stepback prompt
+
+import requests
+
+
+def get_stepback_insight_openai(
+    stepback,
+    api_key,
+    user_model,
+    model_source="openai",
+    creativity=None
+):
+    """
+    Get stepback insight from OpenAI-compatible APIs.
+    Uses direct HTTP requests instead of OpenAI SDK for lighter dependencies.
+    """
+    # Determine the base URL based on model source
+    if model_source == "huggingface":
+        from cat_stack._providers import _detect_huggingface_endpoint
+        base_url = _detect_huggingface_endpoint(api_key, user_model)
+    elif model_source == "huggingface-together":
+        base_url = "https://router.huggingface.co/together/v1"
+    elif model_source == "perplexity":
+        base_url = "https://api.perplexity.ai"
+    elif model_source == "xai":
+        base_url = "https://api.x.ai/v1"
+    else:
+        base_url = "https://api.openai.com/v1"
+
+    endpoint = f"{base_url}/chat/completions"
+
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {api_key}"
+    }
+
+    payload = {
+        "model": user_model,
+        "messages": [{"role": "user", "content": stepback}],
+    }
+
+    if creativity is not None:
+        payload["temperature"] = creativity
+
+    try:
+        response = requests.post(endpoint, headers=headers, json=payload, timeout=120)
+        response.raise_for_status()
+        result = response.json()
+        stepback_insight = result["choices"][0]["message"]["content"]
+
+        return stepback_insight, True
+
+    except Exception as e:
+        print(f"An error occurred during step-back prompting: {e}")
+        return None, False
+
+
+# claude stepback prompt
+
+def get_stepback_insight_anthropic(
+    stepback,
+    api_key,
+    user_model,
+    model_source="anthropic",
+    creativity=None
+):
+    """
+    Get stepback insight from Anthropic Claude.
+
+    Uses direct HTTP requests instead of Anthropic SDK for lighter dependencies.
+    """
+    import requests
+
+    endpoint = "https://api.anthropic.com/v1/messages"
+
+    headers = {
+        "Content-Type": "application/json",
+        "x-api-key": api_key,
+        "anthropic-version": "2023-06-01"
+    }
+
+    payload = {
+        "model": user_model,
+        "max_tokens": 4096,
+        "messages": [{"role": "user", "content": stepback}],
+    }
+
+    if creativity is not None:
+        payload["temperature"] = creativity
+
+    try:
+        response = requests.post(endpoint, headers=headers, json=payload, timeout=120)
+        response.raise_for_status()
+        result = response.json()
+
+        # Parse response - Anthropic returns content as a list
+        content = result.get("content", [])
+        if content and content[0].get("type") == "text":
+            stepback_insight = content[0].get("text", "")
+            return stepback_insight, True
+
+        return None, False
+
+    except Exception as e:
+        print(f"An error occurred during step-back prompting: {e}")
+        return None, False
+    
+# google stepback prompt
+
+def get_stepback_insight_google(
+    stepback,
+    api_key,
+    user_model,
+    model_source="google",
+    creativity=None
+):
+    
+    import requests
+    
+    url = f"https://generativelanguage.googleapis.com/v1beta/models/{user_model}:generateContent?key={api_key}"
+    
+    headers = {
+        "Content-Type": "application/json"
+    }
+    
+    payload = {
+        "contents": [{
+            "parts": [{"text": stepback}],
+
+            **({"generationConfig": {"temperature": creativity}} if creativity is not None else {})
+        }]
+    }
+    
+    try:
+        response = requests.post(url, headers=headers, json=payload)
+        response.raise_for_status()  # Raise error for bad status codes
+        
+        result = response.json()
+        stepback_insight = result['candidates'][0]['content']['parts'][0]['text']
+        
+        return stepback_insight, True
+        
+    except Exception as e:
+        print(f"An error occurred during step-back prompting: {e}")
+        return None, False
+
+# mistral stepback prompt
+
+def get_stepback_insight_mistral(
+    stepback,
+    api_key,
+    user_model,
+    model_source="mistral",
+    creativity=None
+):
+    import requests
+
+    endpoint = "https://api.mistral.ai/v1/chat/completions"
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {api_key}"
+    }
+
+    payload = {
+        "model": user_model,
+        "messages": [{'role': 'user', 'content': stepback}],
+    }
+    if creativity is not None:
+        payload["temperature"] = creativity
+
+    try:
+        response = requests.post(endpoint, headers=headers, json=payload, timeout=120)
+        response.raise_for_status()
+        result = response.json()
+        stepback_insight = result["choices"][0]["message"]["content"]
+
+        return stepback_insight, True
+
+    except Exception as e:
+        print(f"An error occurred during step-back prompting: {e}")
+        return None, False
+    
+# 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}],
+            response_format={"type": "json_object"},
+            **({"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")
+
+        tools = [{
+            "name": "return_categories",
+            "description": "Return categorization results as 0 (not present) or 1 (present) for each category",
+            "input_schema": {
+                "type": "object",
+                "properties": properties,
+                "required": list(properties.keys())  # All categories required
+            }
+        }]
+        
+        final_response = client.messages.create(
+            model=user_model,
+            messages=[{'role': 'user', 'content': step4_filled}],
+            max_tokens=4096,
+            tools=tools,
+            tool_choice={"type": "tool", "name": "return_categories"},
+            **({"temperature": creativity} if creativity is not None else {})
+        )
+        
+        result_dict = final_response.content[0].input
+
+        verified_reply = json.dumps(result_dict)
+        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 {}),
+            **({"thinkingConfig": {"thinkingBudget": thinking_budget}} if thinking_budget 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 {}),
+                **({"thinkingConfig": {"thinkingBudget": thinking_budget}} if thinking_budget 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": {
+                                "responseMimeType": "application/json",
+                                **({"temperature": creativity} if creativity is not None else {}),
+                                **({"thinkingConfig": {"thinkingBudget": thinking_budget}} if thinking_budget 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}],
+            response_format={"type": "json_object"},
+            **({"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
+        
+# openai explore corpus call
+def get_openai_top_n(
+    prompt,
+    user_model,
+    specificity,
+    model_source,
+    api_key,
+    research_question,
+    creativity
+):
+    """
+    Get response from OpenAI API with system message.
+    Uses direct HTTP requests instead of OpenAI SDK for lighter dependencies.
+    """
+    # Determine the base URL based on model source
+    if model_source == "huggingface":
+        from cat_stack._providers import _detect_huggingface_endpoint
+        base_url = _detect_huggingface_endpoint(api_key, user_model)
+    elif model_source == "huggingface-together":
+        base_url = "https://router.huggingface.co/together/v1"
+    elif model_source == "perplexity":
+        base_url = "https://api.perplexity.ai"
+    elif model_source == "xai":
+        base_url = "https://api.x.ai/v1"
+    else:
+        base_url = "https://api.openai.com/v1"
+
+    endpoint = f"{base_url}/chat/completions"
+
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {api_key}"
+    }
+
+    # Build system message
+    if research_question:
+        system_content = (
+            f"You are a helpful assistant that extracts categories from text responses. "
+            f"The specific task is to identify {specificity} categories of responses to a text prompt. "
+            f"The research question is: {research_question}"
+        )
+    else:
+        system_content = "You are a helpful assistant."
+
+    payload = {
+        "model": user_model,
+        "messages": [
+            {"role": "system", "content": system_content},
+            {"role": "user", "content": prompt}
+        ],
+    }
+
+    if creativity is not None:
+        payload["temperature"] = creativity
+
+    response = requests.post(endpoint, headers=headers, json=payload, timeout=120)
+    response.raise_for_status()
+    result = response.json()
+
+    return result["choices"][0]["message"]["content"]
+
+def get_anthropic_top_n(
+    prompt,
+    user_model,
+    model_source,
+    specificity,
+    api_key,
+    research_question,
+    creativity
+):
+    """
+    Get response from Anthropic API with system prompt.
+
+    Uses direct HTTP requests instead of Anthropic SDK for lighter dependencies.
+    """
+    import requests
+
+    endpoint = "https://api.anthropic.com/v1/messages"
+
+    headers = {
+        "Content-Type": "application/json",
+        "x-api-key": api_key,
+        "anthropic-version": "2023-06-01"
+    }
+
+    # Build system prompt
+    if research_question:
+        system_content = (f"You are a helpful assistant that extracts categories from text responses. "
+                        f"The specific task is to identify {specificity} categories of responses to a text prompt. "
+                        f"The research question is: {research_question}")
+    else:
+        system_content = "You are a helpful assistant."
+
+    payload = {
+        "model": user_model,
+        "max_tokens": 4096,
+        "system": system_content,
+        "messages": [{"role": "user", "content": prompt}],
+    }
+
+    if creativity is not None:
+        payload["temperature"] = creativity
+
+    response = requests.post(endpoint, headers=headers, json=payload, timeout=120)
+    response.raise_for_status()
+    result = response.json()
+
+    # Parse response - Anthropic returns content as a list
+    content = result.get("content", [])
+    if content and content[0].get("type") == "text":
+        return content[0].get("text", "")
+
+    return ""