y-translator-cli 0.1.0__py3-none-any.whl

tests/__init__.py

@@ -0,0 +1 @@
+"""Tests for the Y-Translator CLI package""" 

tests/test_cli.py

@@ -0,0 +1,23 @@
+"""Tests for Y-Translator CLI"""
+
+import pytest
+from translator import __version__
+from translator.cli import create_parser
+
+def test_version():
+    """Test that version is set correctly"""
+    assert __version__ is not None
+    assert isinstance(__version__, str)
+    assert len(__version__.split('.')) >= 2
+
+def test_parser_creation():
+    """Test that the argument parser is created properly"""
+    parser = create_parser()
+    args = parser.parse_args([])
+    assert args is not None
+    
+    args = parser.parse_args(['--verbose'])
+    assert args.verbose is True
+    
+    args = parser.parse_args(['--model', 'gpt-3.5-turbo'])
+    assert args.model == 'gpt-3.5-turbo' 

translator/__init__.py

@@ -0,0 +1,6 @@
+"""
+Y-Translator CLI
+A command-line translator between English and Chinese powered by AI
+"""
+
+__version__ = "0.1.0" 

translator/agent.py

@@ -0,0 +1,39 @@
+"""AI agent functionality for Y-Translator CLI"""
+
+from agno.agent import Agent
+from agno.models.openai.like import OpenAILike
+from .config import Config
+
+class TranslatorAgent:
+    """Translator AI Agent wrapper"""
+    
+    def __init__(self, config: Config):
+        """Initialize the agent with configuration"""
+        self.config = config
+        self.model = OpenAILike(
+            id=config.model,
+            base_url=config.api_base,
+            api_key=config.api_key,
+        )
+        
+        self.agent = Agent(
+            instructions=[
+                "You are only a translator",
+                "You only translate user input to English, when user input is Chinese",
+                "You only translate user input to Chinese, when user input is English",
+                "You don't answer any questions",
+            ],
+            markdown=True,
+            model=self.model,
+            stream=True,
+            add_history_to_messages=True,
+            num_history_responses=5,
+            debug_mode=config.debug,
+        )
+    
+    def process_query(self, query: str, stream: bool = True) -> None:
+        """Process a single query and print the response"""
+        if not query.strip():
+            return
+            
+        self.agent.print_response(query, stream=stream) 

translator/cli.py

@@ -0,0 +1,97 @@
+"""Command-line interface for Y-Translator CLI"""
+
+import argparse
+import sys
+from prompt_toolkit import PromptSession
+from prompt_toolkit.history import InMemoryHistory
+from . import __version__
+from .config import Config
+from .agent import TranslatorAgent
+
+def create_parser() -> argparse.ArgumentParser:
+    """Create command line argument parser"""
+    parser = argparse.ArgumentParser(
+        description="Y-Translator CLI - AI-powered English-Chinese translator"
+    )
+    parser.add_argument(
+        "-v", "--version",
+        action="version",
+        version=f"Y-Translator CLI version {__version__}"
+    )
+    parser.add_argument(
+        "--verbose",
+        action="store_true",
+        help="Enable debug mode"
+    )
+    parser.add_argument(
+        "--model",
+        help="AI model to use (default: gpt-4)",
+        default=None
+    )
+    parser.add_argument(
+        "--api-key",
+        help="OpenAI API key",
+        default=None
+    )
+    parser.add_argument(
+        "--api-base",
+        help="Custom API base URL",
+        default=None
+    )
+    parser.add_argument(
+        "-n", "--no-stream",
+        action="store_true",
+        help="Disable streaming mode"
+    )
+    return parser
+
+def run_interactive_session(agent: TranslatorAgent, stream: bool = True):
+    """Run an interactive session with the agent"""
+    history = InMemoryHistory()
+    session = PromptSession(history=history)
+    
+    print("\n" + "="*60)
+    print(f"🚀 \033[1;36mY-Translator CLI v{__version__}\033[0m - \033[1mAI-powered Translator\033[0m")
+    print("="*60)
+    print("\033[1;32m✓\033[0m English to Chinese translation")
+    print("\033[1;32m✓\033[0m Chinese to English translation")
+    print("\033[1;32m✓\033[0m Type 'exit' or press Ctrl+C to quit")
+    print("")
+    print("\033[1;33mType your text to translate and press Enter:\033[0m")
+    
+    while True:
+        try:
+            query = session.prompt('>>> ')
+
+            if query.lower() in ("exit", "quit", "退出"):
+                print("\n\033[1;32mGoodbye! Have a great day!\033[0m")
+                break
+
+            agent.process_query(query, stream=stream)
+
+        except KeyboardInterrupt:
+            print("\n\033[1;32mExiting...\033[0m")
+            break
+        except Exception as e:
+            print(f"\033[1;31mError: {e}\033[0m")
+
+def main():
+    """Main entry point for Y-Translator CLI"""
+    parser = create_parser()
+    args = parser.parse_args()
+    
+    try:
+        config = Config.from_args(args)
+        agent = TranslatorAgent(config)
+        run_interactive_session(agent, stream=not args.no_stream)
+    except KeyboardInterrupt:
+        print("\n\033[1;32mExiting...\033[0m")
+        sys.exit(0)
+    except Exception as e:
+        print(f"\033[1;31mError: {e}\033[0m", file=sys.stderr)
+        if args.verbose:
+            raise
+        sys.exit(1)
+
+if __name__ == "__main__":
+    main() 

translator/config.py

@@ -0,0 +1,23 @@
+"""Configuration management for Y-Translator CLI"""
+
+import os
+from dataclasses import dataclass
+from typing import Optional
+
+@dataclass
+class Config:
+    """Configuration settings for Y-Translator CLI"""
+    api_key: str = os.getenv("AI_API_KEY", "")
+    model: str = os.getenv("AI_MODEL", "gpt-4")
+    api_base: str = os.getenv("AI_API_BASE", "https://api.openai.com/v1")
+    debug: bool = False
+    
+    @classmethod
+    def from_args(cls, args) -> 'Config':
+        """Create config from command line arguments"""
+        return cls(
+            api_key=args.api_key or os.getenv("AI_API_KEY", ""),
+            model=args.model or os.getenv("AI_MODEL", "gpt-4"),
+            api_base=args.api_base or os.getenv("AI_API_BASE", "https://api.openai.com/v1"),
+            debug=args.verbose,
+        ) 

y_translator_cli-0.1.0.dist-info/METADATA

@@ -0,0 +1,117 @@
+Metadata-Version: 2.4
+Name: y-translator-cli
+Version: 0.1.0
+Summary: An AI-powered English-Chinese translator for the command line
+Home-page: https://github.com/yourusername/translator
+Author: Yang
+Author-email: your.email@example.com
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: prompt_toolkit>=3.0.0
+Requires-Dist: agno>=0.1.0
+Requires-Dist: python-dotenv>=0.19.0
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# Y-Translator CLI
+
+An AI-powered command-line translator that converts between English and Chinese.
+
+## Installation
+
+```bash
+pip install y-translator-cli
+```
+
+## Usage
+
+Start the translator:
+```bash
+y-translator
+```
+
+Options:
+- `-h, --help`: Show help message
+- `-v, --version`: Show version information
+- `--verbose`: Enable debug mode
+- `--model MODEL`: Specify AI model to use (default: gpt-4)
+- `--api-key KEY`: Set OpenAI API key
+- `--api-base URL`: Set custom API base URL
+- `-n, --no-stream`: Disable streaming mode
+
+## Environment Variables
+
+You can set the following environment variables:
+- `AI_API_KEY`: Your OpenAI API key
+- `AI_MODEL`: AI model to use (default: gpt-4)
+- `AI_API_BASE`: Custom API base URL
+
+## Examples
+
+1. Start the translator:
+```bash
+y-translator
+```
+
+2. Enable debug mode:
+```bash
+y-translator --verbose
+```
+
+3. Use a specific model:
+```bash
+y-translator --model gpt-3.5-turbo
+```
+
+4. Disable streaming output:
+```bash
+y-translator -n
+```
+
+## Development
+
+### Installation for Development
+
+```bash
+# Clone the repository
+git clone https://github.com/yourusername/y-translator.git
+cd y-translator
+
+# Install in development mode
+pip install -e .
+```
+
+### Building the Package
+
+```bash
+# Use build
+python -m build
+```
+
+If building manually, you may want to clean old files first:
+```bash
+# Clean before building
+rm -rf dist/ build/ *.egg-info/ __pycache__/ .pytest_cache/
+find . -name "*.pyc" -delete
+```
+
+## License
+
+MIT License 

y_translator_cli-0.1.0.dist-info/RECORD

@@ -0,0 +1,12 @@
+tests/__init__.py,sha256=lncvqHZnsQZaPJFvF2mXv3wnnMN9ibTXP9yTIG1YpyA,45
+tests/test_cli.py,sha256=N_xYgp_yghMtYKH4wkKBvMH4uMZY88Yl4Xzx3vfzp98,674
+translator/__init__.py,sha256=hQDYySj23dHElMAgz1_iMVidoLUo4dkBj0OQmM7JWiM,116
+translator/agent.py,sha256=v9iwvaGUePBRxDq4ylbS0R0ojrcT7mt5ckrhqrpqoHc,1301
+translator/cli.py,sha256=JGbs0V9qhXBSGNucJmgD4cUYjNbQyTB-SgCfk75h-BY,2899
+translator/config.py,sha256=_rlQO2hSMtc_HtaHl7F7I6mmLLk1yc19BfYMTZjdgk8,802
+y_translator_cli-0.1.0.dist-info/licenses/LICENSE,sha256=ERLNVd6wQ_wrpxs7LVks0YjnHhwLmW9SsYJUYFy0ukk,1061
+y_translator_cli-0.1.0.dist-info/METADATA,sha256=pevtt19x6QXWGLb-yfTHejjrXWE8am_hqS87dmDSLLs,2489
+y_translator_cli-0.1.0.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+y_translator_cli-0.1.0.dist-info/entry_points.txt,sha256=3O97fL4RZwoxln34xI9uOkVObQUb_3VcJhDIbEHGOew,51
+y_translator_cli-0.1.0.dist-info/top_level.txt,sha256=wPTGyikl7K0psWsSRLzYZqDdal4kcBKPTdbT2BwQca0,17
+y_translator_cli-0.1.0.dist-info/RECORD,,

y_translator_cli-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (78.1.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

y_translator_cli-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+translator = translator.cli:main

y_translator_cli-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2023 Yang
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE. 

y_translator_cli-0.1.0.dist-info/top_level.txt

@@ -0,0 +1,2 @@
+tests
+translator