prompt-llm 0.1.0__tar.gz

prompt_llm-0.1.0/.gitignore

@@ -0,0 +1,4 @@
+build/*
+mycli/*
+__pycache__
+dist

prompt_llm-0.1.0/LICENCE

@@ -0,0 +1,19 @@
+Copyright (c) 2018 The Python Packaging Authority
+
+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.

prompt_llm-0.1.0/PKG-INFO

@@ -0,0 +1,51 @@
+Metadata-Version: 2.3
+Name: prompt_llm
+Version: 0.1.0
+Summary: A CLI tool to prompt LLMs
+Project-URL: Homepage, https://github.com/reego-software/prompt_llm
+Project-URL: Issues, https://github.com/reego-software/prompt_llm/issues
+Author-email: Hasan YILMAZ <yilmazhasan111@gmail.com>
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.8
+Requires-Dist: langchain-core
+Requires-Dist: langchain-mistralai
+Requires-Dist: langchain-openai
+Requires-Dist: typer
+Description-Content-Type: text/markdown
+
+# prompt_llm
+
+## Version
+0.1.0
+
+- Prompt LLMs via cli
+
+## Installation
+
+1. Clone the repo
+
+2. Install
+- Local:
+    - `pip install --editable .`
+- PyPi:
+    - `pip install prompt_llm`
+
+3. Install auto-completion:
+- `prompt_llm --install-completion`
+- `source ~/.bashrc`
+
+## Config:
+- Supported configs: `--system`, `--temperature`, `--api-key`
+
+### Set Config
+- `add_config  --api-key "<api_key>" --system "<system_messgae>" --temperature <temperature>`
+
+### Remove Config:
+
+- `prompt_llm rm-config temperature`
+
+## Run cli
+- `prompt_llm openai "Tell me something"`
+

prompt_llm-0.1.0/README.md

@@ -0,0 +1,34 @@
+# prompt_llm
+
+## Version
+0.1.0
+
+- Prompt LLMs via cli
+
+## Installation
+
+1. Clone the repo
+
+2. Install
+- Local:
+    - `pip install --editable .`
+- PyPi:
+    - `pip install prompt_llm`
+
+3. Install auto-completion:
+- `prompt_llm --install-completion`
+- `source ~/.bashrc`
+
+## Config:
+- Supported configs: `--system`, `--temperature`, `--api-key`
+
+### Set Config
+- `add_config  --api-key "<api_key>" --system "<system_messgae>" --temperature <temperature>`
+
+### Remove Config:
+
+- `prompt_llm rm-config temperature`
+
+## Run cli
+- `prompt_llm openai "Tell me something"`
+

prompt_llm-0.1.0/pyproject.toml

@@ -0,0 +1,29 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "prompt_llm"
+dynamic = ["version"]
+dependencies = ["typer", "langchain-core", "langchain-openai", "langchain-mistralai"]
+authors = [
+  { name="Hasan YILMAZ", email="yilmazhasan111@gmail.com" },
+]
+description = "A CLI tool to prompt LLMs"
+readme = "README.md"
+requires-python = ">=3.8"
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+]
+
+[project.urls]
+Homepage = "https://github.com/reego-software/prompt_llm"
+Issues = "https://github.com/reego-software/prompt_llm/issues"
+
+[project.scripts]
+prompt_llm = "prompt_llm.cli:app"
+
+[tool.hatch.version]
+path = "src/prompt_llm/__init__.py"

prompt_llm-0.1.0/requirements.txt

@@ -0,0 +1,58 @@
+annotated-types==0.7.0
+anyio==4.7.0
+build==1.2.2.post1
+certifi==2024.8.30
+charset-normalizer==3.4.0
+click==8.1.7
+distro==1.9.0
+docutils==0.21.2
+filelock==3.16.1
+fsspec==2024.10.0
+h11==0.14.0
+httpcore==1.0.7
+httpx==0.28.1
+httpx-sse==0.4.0
+huggingface-hub==0.26.5
+idna==3.10
+jaraco.classes==3.4.0
+jaraco.context==6.0.1
+jaraco.functools==4.1.0
+jiter==0.8.0
+jsonpatch==1.33
+jsonpointer==3.0.0
+keyring==25.5.0
+langchain-core==0.3.22
+langchain-mistralai==0.2.3
+langchain-openai==0.2.11
+langsmith==0.1.147
+markdown-it-py==3.0.0
+mdurl==0.1.2
+more-itertools==10.5.0
+nh3==0.2.19
+openai==1.57.0
+orjson==3.10.12
+packaging==24.2
+pkginfo==1.12.0
+pydantic==2.10.3
+pydantic_core==2.27.1
+Pygments==2.18.0
+pyproject_hooks==1.2.0
+PyYAML==6.0.2
+readme_renderer==44.0
+regex==2024.11.6
+requests==2.32.3
+requests-toolbelt==1.0.0
+rfc3986==2.0.0
+rich==13.9.4
+setuptools==75.1.0
+shellingham==1.5.4
+sniffio==1.3.1
+tenacity==9.0.0
+tiktoken==0.8.0
+tokenizers==0.21.0
+tqdm==4.67.1
+twine==6.0.1
+typer==0.15.1
+typing_extensions==4.12.2
+urllib3==2.2.3
+wheel==0.44.0

prompt_llm-0.1.0/src/prompt_llm/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.0"

prompt_llm-0.1.0/src/prompt_llm/cli.py

@@ -0,0 +1,68 @@
+#!/usr/bin/env python
+
+import typer
+import json
+import os
+
+from .utils import add_to_config, remove_from_config, load_config
+from .langchain_util import call
+
+app = typer.Typer()
+
+@app.command()
+def add_config(api_key: str='', system: str = '', temperature:float = None):
+    if api_key:
+        add_to_config('api_key', api_key)
+
+    if system:
+        add_to_config('system', system)
+    
+    if temperature is not None:
+        add_to_config('temperature', temperature)
+
+@app.command()
+def rm_config(key: str):    
+    remove_from_config(key)
+
+
+@app.command()
+async def mistralai(prompt: str):
+    app_config = load_config()
+
+    if prompt:
+        app_config["provider"] = "mistralai"
+        app_config["model"] = "mistral-large-latest"
+
+        response = await call(prompt, app_config)
+
+        print(response)
+
+@app.command()
+def openai(prompt: str):
+    """Tool to prompt openai"""
+    app_config = load_config()
+
+    if prompt:
+        app_config["provider"] = "openai"
+        app_config["model"] = "gpt-4o"
+
+        response = call(prompt, app_config)
+
+        print(response.content)
+
+
+@app.command()
+def version():
+    """Retrieve the version from the __init__.py file."""
+    package_dir = os.path.dirname(os.path.abspath(__file__))
+    init_file = os.path.join(package_dir, '__init__.py')
+
+    with open(init_file, 'r') as file:
+        for line in file:
+            if line.startswith('__version__'):
+                # Extract version from the line
+                version = line.split('=')[-1].strip().strip('"').strip("'")
+                print(version)
+
+if __name__ == "__main__":
+    app()

prompt_llm-0.1.0/src/prompt_llm/langchain_util.py

@@ -0,0 +1,31 @@
+import os
+from langchain_openai import ChatOpenAI
+from langchain_mistralai import ChatMistralAI
+from langchain_core.messages import HumanMessage, AIMessage, SystemMessage
+from .models import Configuration
+
+def get_model_to_invoke(config: dict):
+    model, api_key, provider = config["model"], config["api_key"], config["provider"]
+    model_to_invoke = None
+
+    if provider == 'openai':
+        os.environ["OPENAI_API_KEY"] = api_key
+        model_to_invoke = ChatOpenAI(model=model)
+    elif provider == 'mistralai':
+        os.environ["MISTRAL_API_KEY"] = api_key
+        model_to_invoke = ChatMistralAI(model=model)
+
+    return model_to_invoke
+
+
+def call(prompt: str, config: dict):
+    model_to_invoke = get_model_to_invoke(config)
+
+    # Messages
+    messages_langchain = []
+    messages_langchain.append(SystemMessage(content=config["system"]))
+    messages_langchain.append(HumanMessage(content=prompt))
+
+    response = model_to_invoke.invoke(messages_langchain)
+
+    return response

prompt_llm-0.1.0/src/prompt_llm/models.py

@@ -0,0 +1,7 @@
+from pydantic import BaseModel, Field
+from typing import List, Dict
+
+class Configuration(BaseModel):
+    api_key: str
+    model: str
+    provider: str

prompt_llm-0.1.0/src/prompt_llm/utils.py

@@ -0,0 +1,48 @@
+import json
+from pathlib import Path
+
+config_dir = Path.home() / ".prompt_llm"
+config_file = config_dir / "config.json"
+
+def add_to_config(key:str, value: str):
+    """Set config such as api_key"""
+    # Ensure the directory exists
+    config_dir.mkdir(parents=True, exist_ok=True)
+
+    # Load existing config if it exists
+    if config_file.exists():
+        with open(config_file, "r") as file:
+            config_data = json.load(file)
+    else:
+        config_data = {}
+
+    # Add or update the API key
+    config_data[key] = value
+
+    # Save the updated config
+    with open(config_file, "w") as file:
+        json.dump(config_data, file, indent=4)
+
+def remove_from_config(key:str):
+    """Remove config such as api_key"""
+    # Ensure the directory exists
+    config_dir.mkdir(parents=True, exist_ok=True)
+
+    # Load existing config if it exists
+    if config_file.exists():
+        with open(config_file, "r") as file:
+            config_data = json.load(file)
+            if key in config_data:
+                del config_data[key]
+
+    # Save the updated config
+    with open(config_file, "w") as file:
+        json.dump(config_data, file, indent=4)
+
+def load_config():
+    # Load existing config if it exists
+    if config_file.exists():
+        with open(config_file, "r") as file:
+            app_config = json.load(file)
+
+    return app_config