webscout 8.2.8__py3-none-any.whl → 8.2.9__py3-none-any.whl

webscout/Provider/TwoAI.py

@@ -1,8 +1,11 @@
 from curl_cffi.requests import Session
 from curl_cffi import CurlError
 import json
+import base64
+import time
 from typing import Any, Dict, Optional, Generator, Union
 import re  # Import re for parsing SSE
+import urllib.parse
 
 from webscout.AIutel import Optimizers
 from webscout.AIutel import Conversation
@@ -10,22 +13,153 @@ from webscout.AIutel import AwesomePrompts, sanitize_stream # Import sanitize_st
 from webscout.AIbase import Provider
 from webscout import exceptions
 from webscout.litagent import LitAgent
+from webscout.Extra.tempmail import get_random_email
 
 
 class TwoAI(Provider):
     """
     A class to interact with the Two AI API (v2) with LitAgent user-agent.
+    SUTRA is a family of large multi-lingual language models (LMLMs) developed by TWO AI.
+    SUTRA's dual-transformer extends the power of both MoE and Dense AI language model architectures,
+    delivering cost-efficient multilingual capabilities for over 50+ languages.
+
+    API keys can be generated using the generate_api_key() method, which uses a temporary email
+    to register for the Two AI service and extract the API key from the confirmation email.
     """
 
     AVAILABLE_MODELS = [
-        "sutra-v2",
-        "sutra-r0"
-
+        "sutra-v2",  # Multilingual AI model for instruction execution and conversational intelligence
+        "sutra-r0",  # Advanced reasoning model for complex problem-solving and deep contextual understanding
     ]
 
+    @staticmethod
+    def generate_api_key() -> str:
+        """
+        Generate a new Two AI API key using a temporary email.
+
+        This method:
+        1. Creates a temporary email using webscout's tempmail module
+        2. Registers for Two AI using the Loops.so newsletter form
+        3. Waits for and extracts the API key from the confirmation email
+
+        Returns:
+            str: The generated API key
+
+        Raises:
+            Exception: If the API key cannot be generated
+        """
+        # Get a temporary email
+        email, provider = get_random_email("tempmailio")
+
+        # Register for Two AI using the Loops.so newsletter form
+        loops_url = "https://app.loops.so/api/newsletter-form/cm7i4o92h057auy1o74cxbhxo"
+
+        # Create a session with appropriate headers
+        session = Session()
+        session.headers.update({
+            'User-Agent': LitAgent().random(),
+            'Content-Type': 'application/x-www-form-urlencoded',
+            'Origin': 'https://www.two.ai',
+            'Referer': 'https://app.loops.so/',
+        })
+
+        # Prepare form data
+        form_data = {
+            'email': email,
+            'userGroup': 'Via Framer',
+            'mailingLists': 'cm8ay9cic00x70kjv0bd34k66'
+        }
+
+        # Send the registration request
+        encoded_data = urllib.parse.urlencode(form_data)
+        response = session.post(loops_url, data=encoded_data, impersonate="chrome120")
+
+        if response.status_code != 200:
+            raise Exception(f"Failed to register for Two AI: {response.status_code} - {response.text}")
+
+        # Wait for the confirmation email and extract the API key
+        max_attempts = 5 
+        attempt = 0
+        api_key = None
+        wait_time = 2 
+
+        while attempt < max_attempts and not api_key:
+            messages = provider.get_messages()
+
+            for message in messages:
+                # Check if this is likely the confirmation email based on subject and sender
+                subject = message.get('subject', '')
+                sender = ''
+
+                # Try to get the sender from different possible fields
+                if 'from' in message:
+                    if isinstance(message['from'], dict):
+                        sender = message['from'].get('address', '')
+                    else:
+                        sender = str(message['from'])
+                elif 'sender' in message:
+                    if isinstance(message['sender'], dict):
+                        sender = message['sender'].get('address', '')
+                    else:
+                        sender = str(message['sender'])
+
+                # Look for keywords in the subject that indicate this is the confirmation email
+                subject_match = any(keyword in subject.lower() for keyword in
+                                   ['welcome', 'confirm', 'verify', 'api', 'key', 'sutra', 'two.ai', 'loops'])
+
+                # Look for keywords in the sender that indicate this is from Two AI or Loops
+                sender_match = any(keyword in sender.lower() for keyword in
+                                  ['two.ai', 'sutra', 'loops.so', 'loops', 'no-reply', 'noreply'])
+
+                is_confirmation = subject_match or sender_match
+
+                if is_confirmation:
+                    pass
+                # Try to get the message content from various possible fields
+                content = None
+
+                # Check for body field (seen in the debug output)
+                if 'body' in message:
+                    content = message['body']
+                # Check for content.text field
+                elif 'content' in message and 'text' in message['content']:
+                    content = message['content']['text']
+                # Check for html field
+                elif 'html' in message:
+                    content = message['html']
+                # Check for text field
+                elif 'text' in message:
+                    content = message['text']
+
+                if not content:
+                    continue
+
+                # Look for the API key pattern in the email content
+                # First, try to find the API key directly
+                api_key_match = re.search(r'sutra_[A-Za-z0-9]{60,70}', content)
+
+                # If not found, try looking for the key with the label
+                if not api_key_match:
+                    key_section_match = re.search(r'🔑 SUTRA API Key\s*([^\s]+)', content)
+                    if key_section_match:
+                        api_key_match = re.search(r'(sutra_[A-Za-z0-9]+)', key_section_match.group(1))
+
+                # If still not found, try a more general pattern
+                if not api_key_match:
+                    api_key_match = re.search(r'sutra_\S+', content)
+
+                if api_key_match:
+                    api_key = api_key_match.group(0)
+                    break
+            if not api_key:
+                attempt += 1
+                time.sleep(wait_time)
+        if not api_key:
+            raise Exception("Failed to get API key from confirmation email")
+        return api_key
+
     def __init__(
         self,
-        api_key: str = None,
         is_conversation: bool = True,
         max_tokens: int = 1024,
         timeout: int = 30,
@@ -35,19 +169,39 @@ class TwoAI(Provider):
         proxies: dict = {},
         history_offset: int = 10250,
         act: str = None,
-        model: str = "sutra-v2",  # Update default model
+        model: str = "sutra-v2",  # Default model
         temperature: float = 0.6,
         system_message: str = "You are a helpful assistant."
     ):
-        """Initializes the TwoAI API client."""
+        """
+        Initializes the TwoAI API client.
+
+        Args:
+            is_conversation: Whether to maintain conversation history.
+            max_tokens: Maximum number of tokens to generate.
+            timeout: Request timeout in seconds.
+            intro: Introduction text for the conversation.
+            filepath: Path to save conversation history.
+            update_file: Whether to update the conversation history file.
+            proxies: Proxy configuration for requests.
+            history_offset: Maximum history length in characters.
+            act: Persona for the conversation.
+            model: Model to use. Must be one of AVAILABLE_MODELS.
+            temperature: Temperature for generation (0.0 to 1.0).
+            system_message: System message to use for the conversation.
+        """
         if model not in self.AVAILABLE_MODELS:
             raise ValueError(f"Invalid model: {model}. Choose from: {self.AVAILABLE_MODELS}")
-        self.url = "https://api.two.app/v2/chat/completions"  # Update API endpoint
+
+        # Always auto-generate API key
+        api_key = self.generate_api_key()
+
+        self.url = "https://api.two.ai/v2/chat/completions"  # API endpoint
         self.headers = {
             'User-Agent': LitAgent().random(),
-            'Accept': 'application/json',  # Keep application/json for request, response is text/event-stream
+            'Accept': 'text/event-stream',  # For streaming responses
             'Content-Type': 'application/json',
-            'X-Session-Token': api_key,
+            'Authorization': f'Bearer {api_key}',  # Using Bearer token authentication
             'Origin': 'https://chat.two.ai',
             'Referer': 'https://api.two.app/'
         }
@@ -64,6 +218,7 @@ class TwoAI(Provider):
         self.model = model
         self.temperature = temperature
         self.system_message = system_message
+        self.api_key = api_key
 
         self.__available_optimizers = (
             method
@@ -96,6 +251,19 @@ class TwoAI(Provider):
         content = delta.get("content")
         return content if isinstance(content, str) else None
 
+    def encode_image(self, image_path: str) -> str:
+        """
+        Encode an image file to base64 string.
+
+        Args:
+            image_path: Path to the image file
+
+        Returns:
+            Base64 encoded string of the image
+        """
+        with open(image_path, "rb") as image_file:
+            return base64.b64encode(image_file.read()).decode('utf-8')
+
     def ask(
         self,
         prompt: str,
@@ -104,6 +272,7 @@ class TwoAI(Provider):
         optimizer: str = None,
         conversationally: bool = False,
         online_search: bool = True,
+        image_path: str = None,
     ) -> Union[Dict[str, Any], Generator]:
         conversation_prompt = self.conversation.gen_complete_prompt(prompt)
         if optimizer:
@@ -112,14 +281,36 @@ class TwoAI(Provider):
             else:
                 raise Exception(f"Optimizer is not one of {self.__available_optimizers}")
 
+        # Prepare messages with image if provided
+        if image_path:
+            # Create a message with image content
+            image_content = {
+                "type": "image_url",
+                "image_url": {
+                    "url": f"data:image/jpeg;base64,{self.encode_image(image_path)}"
+                }
+            }
+            user_message = {
+                "role": "user",
+                "content": [
+                    {"type": "text", "text": conversation_prompt},
+                    image_content
+                ]
+            }
+        else:
+            # Text-only message
+            user_message = {"role": "user", "content": conversation_prompt}
+
+        # Prepare the payload
         payload = {
             "messages": [
                 *([{"role": "system", "content": self.system_message}] if self.system_message else []),
-                {"role": "user", "content": conversation_prompt},
+                user_message
             ],
             "model": self.model,
             "temperature": self.temperature,
             "max_tokens": self.max_tokens_to_sample,
+            "stream": stream,
             "extra_body": {
                 "online_search": online_search,
             }
@@ -208,6 +399,7 @@ class TwoAI(Provider):
         optimizer: str = None,
         conversationally: bool = False,
         online_search: bool = True,
+        image_path: str = None,
     ) -> str:
         effective_stream = stream if stream is not None else True
 
@@ -220,6 +412,7 @@ class TwoAI(Provider):
                 optimizer=optimizer,
                 conversationally=conversationally,
                 online_search=online_search,
+                image_path=image_path,
             )
             for response_dict in gen:
                 yield self.get_message(response_dict) # get_message expects dict
@@ -233,6 +426,7 @@ class TwoAI(Provider):
                 optimizer=optimizer,
                 conversationally=conversationally,
                 online_search=online_search,
+                image_path=image_path,
             )
             return self.get_message(response_dict) # get_message expects dict
 
@@ -244,37 +438,38 @@ class TwoAI(Provider):
 
 
 if __name__ == "__main__":
-    from rich import print
-    import os
-
-    api_key = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VySWQiOiJzanl2OHJtZGxDZDFnQ2hQdGxzZHdxUlVteXkyIiwic291cmNlIjoiRmlyZWJhc2UiLCJpYXQiOjE3NDYxMDY0NjksImV4cCI6MTc0NjEwNzM2OX0.o3fprDgsUJwvwCsWr0HfqmVpSBUthHsxqnopfWhtiYc"
-
-    try:
-        ai = TwoAI(
-            api_key=api_key,
-            timeout=60,
-            model="sutra-r0",
-            system_message="You are an intelligent AI assistant. Be concise and helpful."
-        )
-
-        response_stream = ai.chat("write me a poem about AI", stream=True, online_search=True)
-        full_stream_response = ""
-        for chunk in response_stream:
-            print(chunk, end="", flush=True)
-            full_stream_response += chunk
-        print("\n[bold green]Stream Test Complete.[/bold green]\n")
-
-        # Optional: Test non-stream
-        # print("[bold blue]Testing Non-Stream:[/bold blue]")
-        # non_stream_response = ai.chat("What is the capital of France?", stream=False, online_search=False)
-        # print(non_stream_response)
-        # print("[bold green]Non-Stream Test Complete.[/bold green]\n")
-
-
-    except exceptions.FailedToGenerateResponseError as e:
-        print(f"\n[bold red]API Error:[/bold red] {e}")
-    except ValueError as e:
-        print(f"\n[bold red]Configuration Error:[/bold red] {e}")
-    except Exception as e:
-        print(f"\n[bold red]An unexpected error occurred:[/bold red] {e}")
-
+    print("-" * 80)
+    print(f"{'Model':<50} {'Status':<10} {'Response'}")
+    print("-" * 80)
+
+    for model in TwoAI.AVAILABLE_MODELS:
+        try:
+            test_ai = TwoAI(model=model, timeout=60)
+            # Test stream first
+            response_stream = test_ai.chat("Say 'Hello' in one word", stream=True)
+            response_text = ""
+            print(f"\r{model:<50} {'Streaming...':<10}", end="", flush=True)
+            for chunk in response_stream:
+                response_text += chunk
+                # Optional: print chunks as they arrive for visual feedback
+                # print(chunk, end="", flush=True) 
+            
+            if response_text and len(response_text.strip()) > 0:
+                status = "✓"
+                # Clean and truncate response
+                clean_text = response_text.strip() # Already decoded in get_message
+                display_text = clean_text[:50] + "..." if len(clean_text) > 50 else clean_text
+            else:
+                status = "✗ (Stream)"
+                display_text = "Empty or invalid stream response"
+            print(f"\r{model:<50} {status:<10} {display_text}")
+            
+            # Optional: Add non-stream test if needed, but stream test covers basic functionality
+            # print(f"\r{model:<50} {'Non-Stream...':<10}", end="", flush=True)
+            # response_non_stream = test_ai.chat("Say 'Hi' again", stream=False)
+            # if not response_non_stream or len(response_non_stream.strip()) == 0:
+            #      print(f"\r{model:<50} {'✗ (Non-Stream)':<10} Empty non-stream response")
+
+
+        except Exception as e:
+            print(f"\r{model:<50} {'✗':<10} {str(e)}")