webscout 8.3.6__py3-none-any.whl → 8.3.7__py3-none-any.whl

webscout/Provider/OPENAI/copilot.py

@@ -1,321 +0,0 @@
-import json
-import uuid
-import time
-from typing import List, Dict, Optional, Union, Generator, Any
-from urllib.parse import quote
-from curl_cffi.requests import Session, CurlWsFlag
-
-# Import base classes and utility structures
-from .base import OpenAICompatibleProvider, BaseChat, BaseCompletions
-from .utils import (
-    ChatCompletionChunk, ChatCompletion, Choice, ChoiceDelta,
-    ChatCompletionMessage, CompletionUsage, format_prompt, count_tokens
-)
-
-# Attempt to import LitAgent, fallback if not available
-try:
-    from webscout.litagent import LitAgent
-except ImportError:
-    pass
-
-# --- Microsoft Copilot Client ---
-
-class Completions(BaseCompletions):
-    def __init__(self, client: 'Copilot'):
-        self._client = client
-
-    def create(
-        self,
-        *,
-        model: str,
-        messages: List[Dict[str, str]],
-        max_tokens: Optional[int] = None,
-        stream: bool = False,
-        temperature: Optional[float] = None,
-        top_p: Optional[float] = None,
-        timeout: Optional[int] = None,
-        proxies: Optional[dict] = None,
-        **kwargs: Any
-    ) -> Union[ChatCompletion, Generator[ChatCompletionChunk, None, None]]:
-        """
-        Creates a model response for the given chat conversation.
-        Mimics openai.chat.completions.create
-        """
-        # Format the entire conversation using the utility function
-        formatted_prompt = format_prompt(messages, add_special_tokens=True, include_system=True, do_continue=True)
-        
-        request_id = f"chatcmpl-{uuid.uuid4()}"
-        created_time = int(time.time())
-
-        # Handle image if provided
-        image = kwargs.get("image")
-
-        if stream:
-            return self._create_stream(request_id, created_time, model, formatted_prompt, image, timeout=timeout, proxies=proxies)
-        else:
-            return self._create_non_stream(request_id, created_time, model, formatted_prompt, image, timeout=timeout, proxies=proxies)
-
-    def _create_stream(
-        self, request_id: str, created_time: int, model: str, prompt_text: str, image: Optional[bytes] = None,
-        timeout: Optional[int] = None, proxies: Optional[dict] = None
-    ) -> Generator[ChatCompletionChunk, None, None]:
-        original_proxies = self._client.session.proxies
-        if proxies is not None:
-            self._client.session.proxies = proxies
-        else:
-            self._client.session.proxies = {}
-        try:
-            timeout_val = timeout if timeout is not None else self._client.timeout
-            s = self._client.session
-            # Create a new conversation if needed
-            r = s.post(self._client.conversation_url, timeout=timeout_val)
-            if r.status_code != 200:
-                raise RuntimeError(f"Failed to create conversation: {r.text}")
-            conv_id = r.json().get("id")
-
-            # Handle image upload if provided
-            images = []
-            if image:
-                r = s.post(
-                    f"{self._client.url}/c/api/attachments",
-                    headers={"content-type": "image/jpeg"},
-                    data=image,
-                    timeout=timeout_val
-                )
-                if r.status_code != 200:
-                    raise RuntimeError(f"Image upload failed: {r.text}")
-                images.append({"type": "image", "url": r.json().get("url")})
-
-            ws = s.ws_connect(self._client.websocket_url)
-            # Map alias to real model name if needed
-            real_model = Copilot.MODEL_ALIASES.get(model, model)
-            if real_model not in Copilot.AVAILABLE_MODELS:
-                raise RuntimeError(f"Invalid model: {model}. Choose from: {Copilot.AVAILABLE_MODELS}")
-            if real_model == "Smart":
-                mode = "smart"
-            elif "Think" in real_model:
-                mode = "reasoning"
-            else:
-                mode = "chat"
-            ws.send(json.dumps({
-                "event": "send",
-                "conversationId": conv_id,
-                "content": images + [{"type": "text", "text": prompt_text}],
-                "mode": mode,
-                "model": real_model
-            }).encode(), CurlWsFlag.TEXT)
-
-            prompt_tokens = count_tokens(prompt_text)
-            completion_tokens = 0
-            total_tokens = prompt_tokens
-            started = False
-            image_prompt = None
-            while True:
-                try:
-                    msg = json.loads(ws.recv()[0])
-                except Exception:
-                    break
-
-                event = msg.get("event")
-                if event not in ["appendText", "done", "error", "generatingImage", "imageGenerated", "suggestedFollowups", "replaceText"]:
-                    print(f"[Copilot] Unhandled event: {event} | msg: {msg}")
-
-                if event == "appendText":
-                    started = True
-                    content = msg.get("text", "")
-                    content_tokens = count_tokens(content)
-                    completion_tokens += content_tokens
-                    total_tokens = prompt_tokens + completion_tokens
-                    delta = ChoiceDelta(
-                        content=content,
-                        role="assistant"
-                    )
-                    choice = Choice(
-                        index=0,
-                        delta=delta,
-                        finish_reason=None
-                    )
-                    chunk = ChatCompletionChunk(
-                        id=request_id,
-                        choices=[choice],
-                        created=created_time,
-                        model=model
-                    )
-                    yield chunk
-                elif event == "replaceText":
-                    # treat as appendText for OpenAI compatibility
-                    content = msg.get("text", "")
-                    content_tokens = count_tokens(content)
-                    completion_tokens += content_tokens
-                    total_tokens = prompt_tokens + completion_tokens
-                    delta = ChoiceDelta(
-                        content=content,
-                        role="assistant"
-                    )
-                    choice = Choice(
-                        index=0,
-                        delta=delta,
-                        finish_reason=None
-                    )
-                    chunk = ChatCompletionChunk(
-                        id=request_id,
-                        choices=[choice],
-                        created=created_time,
-                        model=model
-                    )
-                    yield chunk
-                elif event == "generatingImage":
-                    image_prompt = msg.get("prompt")
-                elif event == "imageGenerated":
-                    # Yield a chunk with image metadata in the delta (custom extension)
-                    delta = ChoiceDelta(
-                        content=None,
-                        role=None
-                    )
-                    choice = Choice(
-                        index=0,
-                        delta=delta,
-                        finish_reason=None
-                    )
-                    chunk = ChatCompletionChunk(
-                        id=request_id,
-                        choices=[choice],
-                        created=created_time,
-                        model=model
-                    )
-                    chunk.image_url = msg.get("url")
-                    chunk.image_prompt = image_prompt
-                    chunk.image_preview = msg.get("thumbnailUrl")
-                    yield chunk
-                elif event == "suggestedFollowups":
-                    # Yield a chunk with followups in the delta (custom extension)
-                    delta = ChoiceDelta(
-                        content=None,
-                        role=None
-                    )
-                    choice = Choice(
-                        index=0,
-                        delta=delta,
-                        finish_reason=None
-                    )
-                    chunk = ChatCompletionChunk(
-                        id=request_id,
-                        choices=[choice],
-                        created=created_time,
-                        model=model
-                    )
-                    chunk.suggested_followups = msg.get("suggestions")
-                    yield chunk
-                elif event == "done":
-                    delta = ChoiceDelta(
-                        content=None,
-                        role=None
-                    )
-                    choice = Choice(
-                        index=0,
-                        delta=delta,
-                        finish_reason="stop"
-                    )
-                    chunk = ChatCompletionChunk(
-                        id=request_id,
-                        choices=[choice],
-                        created=created_time,
-                        model=model
-                    )
-                    yield chunk
-                    break
-                elif event == "error":
-                    print(f"[Copilot] Error event: {msg}")
-                    raise RuntimeError(f"Copilot error: {msg}")
-
-            ws.close()
-            if not started:
-                raise RuntimeError("No response received from Copilot")
-        except Exception as e:
-            raise RuntimeError(f"Stream error: {e}") from e
-        finally:
-            self._client.session.proxies = original_proxies
-
-    def _create_non_stream(
-        self, request_id: str, created_time: int, model: str, prompt_text: str, image: Optional[bytes] = None,
-        timeout: Optional[int] = None, proxies: Optional[dict] = None
-    ) -> ChatCompletion:
-        result = ""
-        # Pass timeout and proxies to the underlying _create_stream call
-        for chunk in self._create_stream(request_id, created_time, model, prompt_text, image, timeout=timeout, proxies=proxies):
-            if hasattr(chunk, 'choices') and chunk.choices and hasattr(chunk.choices[0], 'delta') and chunk.choices[0].delta.content:
-                result += chunk.choices[0].delta.content
-
-        # Create the message object
-        message = ChatCompletionMessage(
-            role="assistant",
-            content=result
-        )
-
-        # Create the choice object
-        choice = Choice(
-            index=0,
-            message=message,
-            finish_reason="stop"
-        )
-
-        # Estimate token usage using count_tokens
-        prompt_tokens = count_tokens(prompt_text)
-        completion_tokens = count_tokens(result)
-        total_tokens = prompt_tokens + completion_tokens
-
-        # Create usage object
-        usage = CompletionUsage(
-            prompt_tokens=prompt_tokens,
-            completion_tokens=completion_tokens,
-            total_tokens=total_tokens
-        )
-
-        # Create the completion object
-        completion = ChatCompletion(
-            id=request_id,
-            choices=[choice],
-            created=created_time,
-            model=model,
-            usage=usage
-        )
-
-        return completion
-
-class Chat(BaseChat):
-    def __init__(self, client: 'Copilot'):
-        self.completions = Completions(client)
-
-class Copilot(OpenAICompatibleProvider):
-    
-    url = "https://copilot.microsoft.com"
-    conversation_url = f"{url}/c/api/conversations"
-    websocket_url = "wss://copilot.microsoft.com/c/api/chat?api-version=2"
-
-    AVAILABLE_MODELS = ["Copilot", "Think Deeper", "Smart"]
-    MODEL_ALIASES = {
-        "gpt-4o": "Copilot",
-        "o4-mini": "Think Deeper",
-        "gpt-5": "Smart",
-
-    }
-
-    def __init__(self, browser: str = "chrome", tools: Optional[List] = None, **kwargs):
-        self.timeout = 900
-        self.session = Session(impersonate=browser)
-        self.session.proxies = {}
-
-        # Initialize tools
-        self.available_tools = {}
-        if tools:
-            self.register_tools(tools)
-            
-        # Set up the chat interface
-        self.chat = Chat(self)
-
-    @property
-    def models(self):
-        class _ModelList:
-            def list(inner_self):
-                return self.AVAILABLE_MODELS
-        return _ModelList()

webscout/Provider/OPENAI/gptoss.py

@@ -1,288 +0,0 @@
-import requests
-import json
-import time
-import uuid
-from typing import List, Dict, Optional, Union, Generator, Any
-
-from webscout.Provider.OPENAI.base import OpenAICompatibleProvider, BaseChat, BaseCompletions
-from webscout.Provider.OPENAI.utils import (
-    ChatCompletionChunk, ChatCompletion, Choice, ChoiceDelta,
-    ChatCompletionMessage, CompletionUsage, format_prompt, count_tokens
-)
-
-try:
-    from webscout.litagent import LitAgent
-except ImportError:
-    pass
-
-class Completions(BaseCompletions):
-    def __init__(self, client: 'GptOss'):
-        self._client = client
-
-    def create(
-        self,
-        *,
-        model: str,
-        messages: List[Dict[str, str]],
-        max_tokens: Optional[int] = 600,
-        stream: bool = False,
-        temperature: Optional[float] = None,
-        top_p: Optional[float] = None,
-        timeout: Optional[int] = None,
-        proxies: Optional[Dict[str, str]] = None,
-        **kwargs: Any
-    ) -> Union[ChatCompletion, Generator[ChatCompletionChunk, None, None]]:
-        # Format messages into conversation prompt
-        conversation_prompt = format_prompt(messages, add_special_tokens=False, do_continue=True)
-        
-        # Count tokens for usage tracking
-        prompt_tokens = count_tokens(conversation_prompt)
-        
-        payload = {
-            "op": "threads.create",
-            "params": {
-                "input": {
-                    "text": conversation_prompt,
-                    "content": [{"type": "input_text", "text": conversation_prompt}],
-                    "quoted_text": "",
-                    "attachments": []
-                }
-            }
-        }
-        
-        request_id = f"chatcmpl-{uuid.uuid4()}"
-        created_time = int(time.time())
-        
-        if stream:
-            return self._create_stream(request_id, created_time, model, payload, timeout, proxies, prompt_tokens)
-        else:
-            return self._create_non_stream(request_id, created_time, model, payload, timeout, proxies, prompt_tokens)
-
-    def _create_stream(
-        self, request_id: str, created_time: int, model: str, payload: Dict[str, Any],
-        timeout: Optional[int] = None, proxies: Optional[Dict[str, str]] = None,
-        prompt_tokens: int = 0
-    ) -> Generator[ChatCompletionChunk, None, None]:
-        try:
-            response = self._client.session.post(
-                self._client.base_url,
-                headers=self._client.headers,
-                json=payload,
-                stream=True,
-                timeout=timeout or self._client.timeout,
-                proxies=proxies
-            )
-            response.raise_for_status()
-            
-            completion_tokens = 0
-            total_tokens = prompt_tokens
-            
-            for line in response.iter_lines(decode_unicode=True):
-                if line and line.startswith("data: "):
-                    json_str = line[6:]
-                    if json_str == "[DONE]":
-                        break
-                    try:
-                        data = json.loads(json_str)
-                        
-                        # Extract content from GptOss response format
-                        content = None
-                        if (data.get('type') == 'thread.item_updated' and 
-                            data.get('update', {}).get('type') == 'assistant_message.content_part.text_delta'):
-                            content = data.get('update', {}).get('delta')
-                        
-                        if content:
-                            # Count tokens in the content chunk
-                            chunk_tokens = count_tokens(content)
-                            completion_tokens += chunk_tokens
-                            total_tokens = prompt_tokens + completion_tokens
-                            
-                            delta = ChoiceDelta(
-                                content=content,
-                                role="assistant"
-                            )
-                            choice = Choice(
-                                index=0,
-                                delta=delta,
-                                finish_reason=None
-                            )
-                            chunk = ChatCompletionChunk(
-                                id=request_id,
-                                choices=[choice],
-                                created=created_time,
-                                model=model
-                            )
-                            chunk.usage = {
-                                "prompt_tokens": prompt_tokens,
-                                "completion_tokens": completion_tokens,
-                                "total_tokens": total_tokens,
-                                "estimated_cost": None
-                            }
-                            yield chunk
-                    except json.JSONDecodeError:
-                        continue
-            
-            # Final chunk with finish_reason="stop"
-            delta = ChoiceDelta(content=None, role=None)
-            choice = Choice(index=0, delta=delta, finish_reason="stop")
-            chunk = ChatCompletionChunk(
-                id=request_id,
-                choices=[choice],
-                created=created_time,
-                model=model
-            )
-            chunk.usage = {
-                "prompt_tokens": prompt_tokens,
-                "completion_tokens": completion_tokens,
-                "total_tokens": total_tokens,
-                "estimated_cost": None
-            }
-            yield chunk
-            
-        except Exception as e:
-            print(f"Error during GptOss stream request: {e}")
-            raise IOError(f"GptOss request failed: {e}") from e
-
-    def _create_non_stream(
-        self, request_id: str, created_time: int, model: str, payload: Dict[str, Any],
-        timeout: Optional[int] = None, proxies: Optional[Dict[str, str]] = None,
-        prompt_tokens: int = 0
-    ) -> ChatCompletion:
-        try:
-            response = self._client.session.post(
-                self._client.base_url,
-                headers=self._client.headers,
-                json=payload,
-                stream=True,  # GptOss API is event-stream only
-                timeout=timeout or self._client.timeout,
-                proxies=proxies
-            )
-            response.raise_for_status()
-            
-            # Collect all chunks to form complete response
-            full_content = ""
-            completion_tokens = 0
-            
-            for line in response.iter_lines(decode_unicode=True):
-                if line and line.startswith("data: "):
-                    json_str = line[6:]
-                    if json_str == "[DONE]":
-                        break
-                    try:
-                        data = json.loads(json_str)
-                        
-                        # Extract content from GptOss response format
-                        if (data.get('type') == 'thread.item_updated' and 
-                            data.get('update', {}).get('type') == 'assistant_message.content_part.text_delta'):
-                            content = data.get('update', {}).get('delta')
-                            if content:
-                                full_content += content
-                    except json.JSONDecodeError:
-                        continue
-            
-            # Count tokens in the complete response
-            completion_tokens = count_tokens(full_content)
-            total_tokens = prompt_tokens + completion_tokens
-            
-            message = ChatCompletionMessage(
-                role="assistant",
-                content=full_content
-            )
-            choice = Choice(
-                index=0,
-                message=message,
-                finish_reason="stop"
-            )
-            usage = CompletionUsage(
-                prompt_tokens=prompt_tokens,
-                completion_tokens=completion_tokens,
-                total_tokens=total_tokens
-            )
-            completion = ChatCompletion(
-                id=request_id,
-                choices=[choice],
-                created=created_time,
-                model=model,
-                usage=usage
-            )
-            return completion
-            
-        except Exception as e:
-            print(f"Error during GptOss non-stream request: {e}")
-            raise IOError(f"GptOss request failed: {e}") from e
-
-class Chat(BaseChat):
-    def __init__(self, client: 'GptOss'):
-        self.completions = Completions(client)
-
-class GptOss(OpenAICompatibleProvider):
-    AVAILABLE_MODELS = ["gpt-oss-20b", "gpt-oss-120b"]
-    
-    def __init__(
-        self, 
-        browser: str = "chrome", 
-        api_key: str = None,
-        model: str = "gpt-oss-120b",
-        reasoning_effort: str = "high",
-        timeout: int = 30,
-        **kwargs
-    ):
-        super().__init__(api_key=api_key, **kwargs)
-        self.timeout = timeout
-        self.base_url = "https://api.gpt-oss.com/chatkit"
-        self.model = model if model in self.AVAILABLE_MODELS else self.AVAILABLE_MODELS[0]
-        self.reasoning_effort = reasoning_effort
-        self.session = requests.Session()
-        
-        # Generate headers using LitAgent
-        try:
-            agent = LitAgent()
-            fingerprint = agent.generate_fingerprint(browser)
-            self.headers = {
-                "Accept": "text/event-stream",
-                "Accept-Encoding": fingerprint.get("accept_encoding", "gzip, deflate, br"),
-                "Accept-Language": fingerprint.get("accept_language", "en-US,en;q=0.9"),
-                "Content-Type": "application/json",
-                "Cache-Control": "no-cache",
-                "Connection": "keep-alive",
-                "Pragma": "no-cache",
-                "User-Agent": fingerprint.get("user_agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"),
-                "x-reasoning-effort": self.reasoning_effort,
-                "x-selected-model": self.model,
-                "x-show-reasoning": "true"
-            }
-        except:
-            # Fallback headers if LitAgent fails
-            self.headers = {
-                "Accept": "text/event-stream",
-                "Accept-Encoding": "gzip, deflate, br",
-                "Accept-Language": "en-US,en;q=0.9",
-                "Content-Type": "application/json",
-                "Cache-Control": "no-cache",
-                "Connection": "keep-alive",
-                "Pragma": "no-cache",
-                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36",
-                "x-reasoning-effort": self.reasoning_effort,
-                "x-selected-model": self.model,
-                "x-show-reasoning": "true"
-            }
-        
-        self.session.headers.update(self.headers)
-        self.chat = Chat(self)
-
-    @property
-    def models(self):
-        class _ModelList:
-            def list(inner_self):
-                return type(self).AVAILABLE_MODELS
-        return _ModelList()
-
-if __name__ == "__main__":
-    client = GptOss()
-    response = client.chat.completions.create(
-        model="gpt-oss-120b",
-        messages=[{"role": "user", "content": "Hello, how are you?"}],
-        max_tokens=100,
-        stream=False
-    )
-    print(response)