PikoAi 0.1.0__py3-none-any.whl

llm_interface/llm.py

@@ -0,0 +1,148 @@
+#simple interface for now. will integrate will other models to make orchestration simpler.For 
+# faster tasks you can do with lighter models
+from dotenv import load_dotenv
+import os
+# from groq import Groq # Commented out
+# from openai import OpenAI # Commented out
+import sys
+import json
+import litellm # Added import for litellm
+
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../../')))
+
+from Src.Utils.ter_interface import TerminalInterface
+
+
+# Load environment variables from .env file
+load_dotenv()
+
+# Accessing an environment variable
+
+# from mistralai import Mistral # Commented out
+
+# def get_model_name(): # Commented out as logic moved to LiteLLMInterface.load_config()
+#     config_path = os.path.join(os.path.dirname(__file__), '../../config.json')
+#     with open(config_path, "r") as config_file:
+#         config = json.load(config_file)
+#         return config.get("model_name", "mistral-large-latest")
+
+class LiteLLMInterface:
+    def __init__(self):
+        self.terminal = TerminalInterface()
+        self.model_name = self.load_config()
+
+    def load_config(self):
+        config_path = os.path.join(os.path.dirname(__file__), '../../config.json')
+        try:
+            with open(config_path, "r") as config_file:
+                config = json.load(config_file)
+                return config.get("model_name", "gpt-3.5-turbo") # Default to gpt-3.5-turbo
+        except FileNotFoundError:
+            print(f"Warning: Config file not found at {config_path}. Defaulting model to gpt-3.5-turbo.")
+            return "gpt-3.5-turbo"
+        except json.JSONDecodeError:
+            print(f"Warning: Error decoding JSON from {config_path}. Defaulting model to gpt-3.5-turbo.")
+            return "gpt-3.5-turbo"
+
+    def chat(self, messages):
+        response_content = ""
+        try:
+            stream_response = litellm.completion(
+                model=self.model_name,
+                messages=messages,
+                stream=True,
+            )
+
+            for chunk in stream_response:
+                content = chunk.choices[0].delta.content
+                if content:
+                    self.terminal.process_markdown_chunk(content)
+                    response_content += content
+            
+            self.terminal.flush_markdown()
+            return response_content
+        except Exception as e:
+            # litellm maps exceptions to OpenAI exceptions.
+            # The executor should catch these and handle them.
+            print(f"An error occurred during the API call: {e}")
+            self.terminal.flush_markdown() # Ensure terminal is flushed even on error
+            raise # Re-raise the exception to be caught by the executor
+
+
+# class MistralModel:
+#     def __init__(self):
+#         api_key = os.environ.get('MISTRAL_API_KEY')
+#         self.client = Mistral(api_key=api_key)
+#         self.terminal = TerminalInterface()
+#         self.model_name = get_model_name()
+
+#     def chat(self, messages):
+        
+#         response = ""
+#         stream_response = self.client.chat.stream(
+#             model=self.model_name,
+#             messages=messages,
+#             temperature=0.2,
+#         )
+
+#         for chunk in stream_response:
+#             content = chunk.data.choices[0].delta.content
+#             if content:
+#                 self.terminal.process_markdown_chunk(content)
+#                 response += content
+
+#         self.terminal.flush_markdown()
+#         return response
+
+        
+# class Groqinference:
+#     def __init__(self):
+#         api_key = os.environ.get('GROQ_API_KEY')
+#         self.client = Groq(
+#             api_key=api_key,
+#         )
+#         self.model_name = get_model_name()
+
+#     def chat(self, messages):
+#         chat_completion = self.client.chat.completions.create(
+#             messages=messages,
+#             model=self.model_name,
+#         )
+#         return chat_completion
+
+# class OpenAi:
+#     def __init__(self):
+#         api_key = os.environ.get('OPENAI_API_KEY')
+#         self.client = OpenAI(api_key=api_key)
+#         self.terminal = TerminalInterface()
+#         self.model_name = get_model_name()
+
+#     def chat(self, messages):
+#         response = ""
+#         stream = self.client.chat.completions.create(
+#             model=self.model_name,
+#             messages=messages,
+#             stream=True
+#         )
+
+#         for chunk in stream:
+#             content = chunk.choices[0].delta.content
+#             if content is not None:
+#                 self.terminal.process_markdown_chunk(content)
+#                 response += content
+
+#         self.terminal.flush_markdown()
+#         return response
+
+# for groq 
+# print(chat_completion.choices[0].message.content)
+
+        # # Iterate over the stream and store chunks
+        # for chunk in stream_response:
+        #     content = chunk.data.choices[0].delta.content
+        #     print(content, end="")  # Print in real-time
+        #     full_response += content  # Append to the full response
+
+        # # Now `full_response` contains the complete response
+        # # Perform operations on the complete response
+        # print("\n\nFull response captured:")

pikoai-0.1.0.dist-info/METADATA

@@ -0,0 +1,101 @@
+Metadata-Version: 2.4
+Name: PikoAi
+Version: 0.1.0
+Summary: An AI-powered task automation tool
+Home-page: https://github.com/nihaaaar22/OS-Assistant
+Author: Nihar S
+Author-email: nihar.sr22@gmail.com
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: python-dotenv>=1.0.1
+Requires-Dist: openai>=1.58.1
+Requires-Dist: groq>=0.22.0
+Requires-Dist: requests>=2.32.3
+Requires-Dist: pdfplumber>=0.11.4
+Requires-Dist: beautifulsoup4>=4.13.4
+Requires-Dist: duckduckgo_search>=7.4.2
+Requires-Dist: rich>=13.9.4
+Requires-Dist: mistralai>=1.2.5
+Requires-Dist: click>=8.1.8
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: psutil>=5.9.8
+Requires-Dist: inquirer>=3.1.3
+Requires-Dist: litellm
+Requires-Dist: PyPDF2
+Requires-Dist: python-docx
+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
+
+<p align="center">
+  <img src="https://www.python.org/static/community_logos/python-logo-master-v3-TM.png" alt="Python 3.8+" width="200"/>
+</p>
+
+# TaskAutomator
+
+**Empower your terminal with AI!**
+
+Just type `ocp` and open a chat with your companion, your copilot. It will search the internet for you, scrape information from sites, help you with projects, do research, and much more. Your system just became a lot more capable. You now have a companion.
+
+---
+
+## 🚀 Core Features
+
+- **LLM-Powered Task Automation:** Utilizes LLMs to understand prompts and orchestrate task execution.
+- **Multi-Provider Support:** Seamlessly switch between LLM providers like Mistral, Groq, and OpenAI.
+- **Extensible Tool System:** Equip agents with custom tools to interact with various environments and APIs.
+- **Flexible Execution Modes:**
+  - **Conversational Mode:** Engage in an interactive dialogue to accomplish tasks.
+  - **One-Shot Task Execution:** Directly execute specific tasks with a single command.
+- **User-Friendly CLI:** A command-line interface to manage configurations, tools, and task execution.
+
+---
+
+## 🛠️ Getting Started
+
+### **Prerequisites**
+- ![Python 3.8+](https://img.shields.io/badge/Python-3.8%2B-blue)
+
+### **Installation**
+Install using pip:
+
+```bash
+pip install ocp
+```
+
+---
+
+## ⚡ One-Shot Task Execution
+
+```bash
+ocp --task "Your task description here"
+```
+
+You can set the maximum number of iterations for a task:
+
+```bash
+ocp --task "Your task description here" --max-iter 5
+```
+
+---
+
+## 🤝 Contributing
+
+We welcome contributions! Please feel free to fork the repository, make your changes, and submit a pull request. For major changes, please open an issue first to discuss what you would like to change.
+
+---
+
+## 📄 License
+
+This project is licensed under the terms of the LICENSE file

pikoai-0.1.0.dist-info/RECORD

@@ -0,0 +1,26 @@
+Agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+Agents/Executor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+Agents/Executor/executor.py,sha256=OpaZCTVl9o0WcOM50Y4W-4RR6XClwlY1OTUwwhZxsh8,12516
+Agents/Executor/prompts.py,sha256=wLS3lPAYWjeKF02LzJ8vP5bZ2VQrMJUd4A7rBfl6qSQ,3846
+Env/__init__.py,sha256=KLe7UcNV5L395SxhMwbYGyu7KPrSNaoV_9QJo3mLop0,196
+Env/base_env.py,sha256=ORM6U5qwj7cTuSHFtSmCSsE0cl6pZ28D97CEyyFnucI,1323
+Env/base_executor.py,sha256=awTwJ44CKWV4JO2KUHfHDX0p1Ujw55hlaL5oNYTEW9M,893
+Env/env.py,sha256=I3lOoyBJR5QNq3tWk_HVH6ncRx1vnilRYmj7b7h4jic,114
+Env/js_executor.py,sha256=tEAg5ho8Pa8LzxUbS1Idau8XuJWZZqPiNlvFPDwGkgc,2690
+Env/python_executor.py,sha256=KiQxyxLgdEph1iRBLteqYy6KJGE8ag_jxiYJ_-BEEB4,3221
+Env/shell.py,sha256=gr6czmeuSWtB3xSA9TZN7wnK2BENOuA9zjNttwbxztU,1877
+Tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+Tools/file_task.py,sha256=VUhWq_G-SWvGahQo8PG7TOpElHUW3BGLUabrTdJS89o,12151
+Tools/system_details.py,sha256=7-mTm3CG4NoatHcvcosalEgEcpWlNsCsZ7kuS3y_EmY,2262
+Tools/tool_manager.py,sha256=ulrFDICPfjDJc7BPTUGdEmJX9Umo0Jnsmz2T05z_a2o,1744
+Tools/userinp.py,sha256=vUhEj3y1W1_ZFHqo2xQwvqDyeOg3VsisSKTI0EurUH8,1205
+Tools/web_loader.py,sha256=PyZk2g7WngZT0tCLs9Danx20dYspnaZwy4rlVE9Sx_4,5054
+Tools/web_search.py,sha256=4EGq1VZqfDgG-_yXTd4_Ha1iEUcR-szdlgRV7oFPru4,1259
+llm_interface/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+llm_interface/llm.py,sha256=4ny7punKBXfXfBuYrGpdp7DW2jbhHHksrPkydHSqnxU,5189
+pikoai-0.1.0.dist-info/licenses/LICENSE,sha256=cELUVOboOAderKFp8bdtcM5VyJi61YH1oDbRhOuoQZw,1067
+pikoai-0.1.0.dist-info/METADATA,sha256=68B0uut29umBRzeRX7RWE8wqnWVBNPpNDCcHGeCDcMs,2961
+pikoai-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pikoai-0.1.0.dist-info/entry_points.txt,sha256=QVeDO6N3nO3UScMb2ksusQWPgcVn86vXosgL-8gu6fo,33
+pikoai-0.1.0.dist-info/top_level.txt,sha256=BdpmoBrO5mzJyrLfIpeh-gw0SFY9k8V8f2jsnzGsg9c,31
+pikoai-0.1.0.dist-info/RECORD,,

pikoai-0.1.0.dist-info/WHEEL

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

pikoai-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+piko = cli:cli

pikoai-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 nihaaaar22
+
+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.

pikoai-0.1.0.dist-info/top_level.txt

@@ -0,0 +1,4 @@
+Agents
+Env
+Tools
+llm_interface