openrouter-provider 0.0.5__py3-none-any.whl → 1.0.10__py3-none-any.whl

OpenRouterProvider/Chatbot_manager.py

@@ -1,123 +0,0 @@
-from .Chat_message import *
-from .OpenRouterProvider import *
-from .LLMs import LLMModel
-
-from dotenv import load_dotenv
-import time
-import json
-
-_base_system_prompt = """
-It's [TIME] today.
-You are an intelligent AI. You must follow the system_instruction below, which is provided by the user.
-
-<system_instruction>
-[SYSTEM_INSTRUCTION]
-</system_instruction>
-"""
-
-class Chatbot_manager:
-    def __init__(self, system_prompt:str="", tools:list[tool_model]=[]) -> None:
-        load_dotenv()
-        
-        self._memory: list[Chat_message] = []
-        self.tools: list[tool_model] = tools
-        self.set_system_prompt(prompt=system_prompt)
-        
-    def set_system_prompt(self, prompt: str):
-        m, d, y = time.localtime()[:3]
-
-        system_prompt = _base_system_prompt
-        system_prompt = system_prompt.replace("[TIME]", f"{m}/{d}/{y}")
-        system_prompt = system_prompt.replace("[SYSTEM_INSTRUCTION]", prompt)
-        
-        self._system_prompt = Chat_message(text=system_prompt, role=Role.system)
-        
-    def clear_memory(self):
-        self._memory = []
-        
-    def print_memory(self):
-        print("\n--------------------- Chatbot memory ---------------------")
-        print(f"system   : {self._system_prompt.text}")
-
-        for message in self._memory:
-            role = message.role.value
-            text = message.text.strip()
-            
-            reset_code = "\033[0m"
-            role_str = f"{role.ljust(9)}:"
-            indent = " " * len(role_str)
-            lines = text.splitlines()
-            
-            if role == "user":
-                color_code = "\033[94m"  # blue
-                if lines:
-                    print(f"{color_code}{role_str}{reset_code} {lines[0]}")
-                    for line in lines[1:]:
-                        print(f"{color_code}{indent}{reset_code} {line}")
-                else:
-                    print(f"{color_code}{role_str}{reset_code}")
-            
-            elif role == "assistant":
-                color_code = "\033[92m"  # green
-                if lines:
-                    print(f"{color_code}{role_str}{reset_code} {lines[0]}")
-                    for line in lines[1:]:
-                        print(f"{color_code}{indent}{reset_code} {line}")
-                else:
-                    print(f"{color_code}{role_str}{reset_code}")
-
-            elif role == "tool":
-                color_code = "\033[93m"  # orange
-                print(f"{color_code}{role_str}{reset_code} ", end="")
-                
-                for tool in message.tool_calls:
-                    print(f"{tool.name}({json.loads(tool.arguments)}), ", end="")
-                print()
-                
-            else:
-                color_code = "\033[0m"   # default color
-                print("Print error: The role is invalid.")
-            
-        print("----------------------------------------------------------\n")
-
-    def invoke(self, model: LLMModel, query: Chat_message, tools: list[tool_model]=[], provider:ProviderConfig=None) -> Chat_message:
-        self._memory.append(query)
-        client = OpenRouterProvider()
-        reply = client.invoke(
-            model=model,
-            system_prompt=self._system_prompt,
-            querys=self._memory,
-            tools=self.tools + tools,
-            provider=provider
-        )
-        reply.answeredBy = model
-        self._memory.append(reply)
-
-        if reply.tool_calls:
-            for requested_tool in reply.tool_calls:
-                args = requested_tool.arguments
-                if isinstance(args, str):
-                    args = json.loads(args)
-
-                for tool in (self.tools + tools):
-                    if tool.name == requested_tool.name:
-                        result = tool(**args)
-                        requested_tool.result = result
-                        break
-                else:
-                    print("Tool Not found", requested_tool.name)
-                    return reply
-                    
-            reply = client.invoke(
-                model=model,
-                system_prompt=self._system_prompt,
-                querys=self._memory,
-                tools=self.tools + tools,
-                provider=provider
-            )
-            
-            reply.answeredBy = model
-            self._memory.append(reply)
-
-        return reply
-    

OpenRouterProvider/OpenRouterProvider.py

@@ -1,112 +0,0 @@
-import logging
-from .Chat_message import *
-from .Tool import tool_model
-from .LLMs import *
-
-from openai import OpenAI
-from dotenv import load_dotenv
-import os
-from dataclasses import dataclass, field, asdict
-from typing import List, Optional, Literal
-import json
-
-# エラーのみ表示、詳細なトレースバック付き
-logging.basicConfig(level=logging.ERROR, format="%(asctime)s - %(levelname)s - %(message)s")
-logger = logging.getLogger(__name__)
-
-
-@dataclass
-class ProviderConfig:
-    order: Optional[List[str]] = None
-    allow_fallbacks: bool = None
-    require_parameters: bool = None
-    data_collection: Literal["allow", "deny"] = None
-    only: Optional[List[str]] = None
-    ignore: Optional[List[str]] = None
-    quantizations: Optional[List[str]] = None
-    sort: Optional[Literal["price", "throughput"]] = None
-    max_price: Optional[dict] = None
-
-    def to_dict(self) -> dict:
-        return {k: v for k, v in asdict(self).items() if v is not None}
-
-
-class OpenRouterProvider:
-    def __init__(self) -> None:
-        load_dotenv()
-        api_key = os.getenv("OPENROUTER_API_KEY")
-        if not api_key:
-            logger.error("OPENROUTER_API_KEY is not set in environment variables.")
-        self.client = OpenAI(
-            base_url="https://openrouter.ai/api/v1",
-            api_key=api_key,
-        )
-
-    def make_prompt(self, system_prompt: Chat_message,
-                    querys: list[Chat_message]) -> list[dict]:
-        messages = [{"role": "system", "content": system_prompt.text}]
-
-        for query in querys:
-            if query.role == Role.user:
-                if query.images is None:
-                    messages.append({"role": "user", "content": query.text})
-                else:
-                    content = [{"type": "text", "text": query.text}]
-                    for img in query.images[:50]:
-                        content.append(
-                            {"type": "image_url",
-                             "image_url": {"url": f"data:image/jpeg;base64,{img}"}})
-                    messages.append({"role": "user", "content": content})
-
-            elif query.role == Role.ai or query.role == Role.tool:
-                assistant_msg = {"role": "assistant"}
-                assistant_msg["content"] = query.text or None
-
-                if query.tool_calls:
-                    assistant_msg["tool_calls"] = [
-                        {
-                            "id": str(t.id),
-                            "type": "function",
-                            "function": {
-                                "name": t.name,
-                                "arguments": t.arguments
-                            }
-                        }
-                        for t in query.tool_calls
-                    ]
-                messages.append(assistant_msg)
-
-            for t in query.tool_calls:
-                messages.append({
-                    "role": "tool",
-                    "tool_call_id": str(t.id),
-                    "content": str(t.result)
-                })
-
-        return messages
-
-    def invoke(self, model: LLMModel, system_prompt: Chat_message, querys: list[Chat_message], tools: list[tool_model] = [], provider: ProviderConfig = None) -> Chat_message:
-        try:
-            messages = self.make_prompt(system_prompt, querys)
-
-            tool_defs = [tool.tool_definition for tool in tools] if tools else None
-            provider_dict = provider.to_dict() if provider else None
-
-            response = self.client.chat.completions.create(
-                model=model.name,
-                messages=messages,
-                tools=tool_defs,
-                extra_body={"provider": provider_dict}
-            )
-
-            reply = Chat_message(text=response.choices[0].message.content, role=Role.ai, raw_response=response)
-
-            if response.choices[0].message.tool_calls:
-                reply.role = Role.tool
-                for tool in response.choices[0].message.tool_calls:
-                    reply.tool_calls.append(ToolCall(id=tool.id, name=tool.function.name, arguments=tool.function.arguments))
-            return reply
-
-        except Exception as e:
-            logger.exception(f"An error occurred while invoking the model: {e.__class__.__name__}: {str(e)}")
-            return Chat_message(text="Fail to get response. Please see the error message.", role=Role.ai, raw_response=None)

openrouter_provider-0.0.5.dist-info/METADATA

@@ -1,232 +0,0 @@
-Metadata-Version: 2.4
-Name: openrouter-provider
-Version: 0.0.5
-Summary: This is an unofficial wrapper of OpenRouter.
-Author-email: Keisuke Miyamto <aichiboyhighschool@gmail.com>
-Requires-Python: >=3.7
-Description-Content-Type: text/markdown
-Requires-Dist: annotated-types
-Requires-Dist: anyio
-Requires-Dist: certifi
-Requires-Dist: distro
-Requires-Dist: h11
-Requires-Dist: httpcore
-Requires-Dist: httpx
-Requires-Dist: idna
-Requires-Dist: jiter
-Requires-Dist: openai
-Requires-Dist: pillow
-Requires-Dist: pydantic
-Requires-Dist: pydantic_core
-Requires-Dist: python-dotenv
-Requires-Dist: sniffio
-Requires-Dist: tqdm
-Requires-Dist: typing-inspection
-Requires-Dist: typing_extensions
-
-## Introduction
-
-Welcome to **openrouter-provider**, an unofficial Python wrapper for the OpenRouter API. This library lets you easily integrate with OpenRouter models, manage chat sessions, process images, and call tools within your Python application.
-
-
-## Features
-
-* Simple chat interface with system, user, assistant, and tool roles
-* Automatic image resizing and Base64 encoding
-* Built-in tool decorator for defining custom functions
-
-
-## Installation
-
-### From PyPI
-
-```bash
-pip3 install openrouter-provider
-```
-
-### From Source
-
-```bash
-git clone https://github.com/yourusername/openrouter-provider.git
-cd openrouter-provider
-pip3 install .
-```
-
-
-
-## Configuration
-
-1. Create a `.env` file in your project root.
-2. Add your OpenRouter API key:
-
-   ```bash
-   OPENROUTER_API_KEY=your_api_key_here
-   ```
-
-
-
-## Usage
-
-### Basic chat bot
-Chat history is automatically sent, by Chatbot_manager. If you want to delete chat history, use `clear_memory` method.
-
-```python
-from OpenRouterProvider.Chatbot_manager import Chat_message, Chatbot_manager
-from OpenRouterProvider.LLMs import gpt_4o_mini
-
-# Declare chat bot
-ai = Chatbot_manager(system_prompt="Please answer in English.")
-
-# Send query
-query = Chat_message(text="Introduce yourself, please.")
-response = ai.invoke(model=gpt_4o_mini, query=query)
-print(response.text)
-
-# Send next query. Chatbot_manager automatically handle chat history.
-query = Chat_message(text="Tell me a short story.")
-response = ai.invoke(model=gpt_4o_mini, query=query)
-print(response.text)
-
-# Print all chat history
-ai.print_memory()  
-
-# Delete all chat history
-ai.clear_memory()
-```
-
-### Chat bot with images
-You can use images in the chat.
-
-```python
-from OpenRouterProvider.Chatbot_manager import Chat_message, Chatbot_manager
-from OpenRouterProvider.LLMs import gpt_4o_mini
-from PIL import Image
-
-dog = Image.open("dog.jpg")
-cat = Image.open("cat.jpg")
-
-# Send query with images
-ai = Chatbot_manager(system_prompt="Please answer in English.")
-query = Chat_message(text="What can you see in the images?", images=[dog, cat])
-response = ai.invoke(model=gpt_4o_mini, query=query)
-print(response.text) 
-```
-
-### With tools
-
-Use the `@tool_model` decorator to expose Python functions as callable tools in the chat. Tools are automatically processed by Chat_manager, so you don't need to care it.
-
-```python
-from OpenRouterProvider.Chatbot_manager import Chat_message, Chatbot_manager
-from OpenRouterProvider.LLMs import gpt_4o_mini
-from OpenRouterProvider.Tool import tool_model
-
-@tool_model
-def get_user_info():
-    """
-    Return user's personal info: name, age, and address.
-    """
-    return "name: Alice\nage: 30\naddress: Wonderland"
-
-ai = Chatbot_manager(system_prompt="Please answer in English.", tools=[get_user_info])
-query = Chat_message(text="What is the name, age, address of the user?")
-response = ai.invoke(model=gpt_4o_mini, query=query)
-ai.print_memory()
-```
-
-## Advanced Usage
-### Prebuilt and Custom Model Usage
-
-You can use prebuilt models defined or declare your own custom models easily.
-This library provides many ready-to-use models from OpenAI, Anthropic, Google, and others.
-
-```python
-from OpenRouterProvider.Chatbot_manager import Chat_message, Chatbot_manager
-from OpenRouterProvider.LLMs import gpt_4o, claude_3_7_sonnet
-
-# Use OpenAI GPT-4o
-ai = Chatbot_manager(system_prompt="Please answer in English.")
-query = Chat_message(text="Tell me a joke.")
-response = ai.invoke(model=gpt_4o, query=query)
-print(response.text)
-
-# Use Anthropic Claude 3.7 Sonnet
-query = Chat_message(text="Summarize the story of Hamlet.")
-response = ai.invoke(model=claude_3_7_sonnet, query=query)
-print(response.text)
-```
-
-Available prebuilt models include:
-
-#### **OpenAI**
-
-* `gpt_4o`
-* `gpt_4o_mini`
-* `gpt_4_1`
-* `gpt_4_1_mini`
-* `gpt_4_1_nano`
-* `o4_mini`
-* `o4_mini_high`
-* `o3`
-
-#### **Anthropic**
-
-* `claude_3_7_sonnet`
-* `claude_3_7_sonnet_thinking`
-* `claude_3_5_haiku`
-
-#### **Google**
-
-* `gemini_2_0_flash`
-* `gemini_2_0_flash_free`
-* `gemini_2_5_flash`
-* `gemini_2_5_flash_thinking`
-* `gemini_2_5_pro`
-
-#### **Deepseek**
-
-* `deepseek_v3_free`
-* `deepseek_v3`
-* `deepseek_r1_free`
-* `deepseek_r1`
-
-#### **xAI**
-
-* `grok_3_mini`
-* `grok_3`
-
-#### **Microsoft**
-
-* `mai_ds_r1_free`
-
-#### **Others**
-
-* `llama_4_maverick_free`
-* `llama_4_scout`
-* `mistral_small_3_1_24B_free`
-
-All of them are instances of `LLMModel`, which includes cost and model name settings.
-
-### Using Custom Models
-
-You can define and use your own custom model if it's available on OpenRouter.
-
-```python
-from OpenRouterProvider.Chatbot_manager import Chat_message, Chatbot_manager
-from OpenRouterProvider.LLMs import LLMModel
-
-# Declare a custom model
-my_model = LLMModel(
-    name="my-org/my-custom-model",  # Model name for OpenRouter
-    input_cost=0.5,                 # Optional: cost per 1M input tokens
-    output_cost=2.0                 # Optional: cost per 1M output tokens
-)
-
-# Use the custom model
-ai = Chatbot_manager(system_prompt="Please answer in English.")
-query = Chat_message(text="Explain black holes simply.")
-response = ai.invoke(model=my_model, query=query)
-print(response.text)
-```
-
-You only need to know the model name as used on OpenRouter. `input_cost` and `output_cost` are optional and currently, they are not used in this library. Please wait the future update.

openrouter_provider-0.0.5.dist-info/RECORD

@@ -1,10 +0,0 @@
-__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-OpenRouterProvider/Chat_message.py,sha256=lQd8bFp7OHOgeOrcpcVZMdkV2Mb4reUsv5Ixo6WecYY,4424
-OpenRouterProvider/Chatbot_manager.py,sha256=EpLWhxx7xnRa-q7xqP2Ur9dmYb9Mzv_UF6BChwpcbYk,4357
-OpenRouterProvider/LLMs.py,sha256=-0ELd6fqmdDvsdaPIElRsluiK85-Y6USwvQb2b4M8TA,2607
-OpenRouterProvider/OpenRouterProvider.py,sha256=4k87D5kKNPgRJ-7qMv9oPWY7P2psrNaNOALmyPMuNsw,4397
-OpenRouterProvider/Tool.py,sha256=QeeWOD2oaYjB9tjF-Jvcjd_G_qSUIuKwFgyh20Ne06I,2010
-openrouter_provider-0.0.5.dist-info/METADATA,sha256=_H9lXm0ohRX57GdvJMJNIyn6pyWblQQTDHOWv7EM6GE,5995
-openrouter_provider-0.0.5.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-openrouter_provider-0.0.5.dist-info/top_level.txt,sha256=I5BMEzkQFEnEYTqOY1Ktmnp7r1rrZQyeWdclKyyyHKs,28
-openrouter_provider-0.0.5.dist-info/RECORD,,

openrouter_provider-0.0.5.dist-info/top_level.txt

@@ -1,2 +0,0 @@
-OpenRouterProvider
-__init__