PikoAi 0.1.18__tar.gz → 0.1.19__tar.gz

{pikoai-0.1.18 → pikoai-0.1.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PikoAi
-Version: 0.1.18
+Version: 0.1.19
 Summary: An AI-powered task automation tool
 Home-page: https://github.com/nihaaaar22/OS-Assistant
 Author: Nihar S

{pikoai-0.1.18 → pikoai-0.1.19}/Src/PikoAi.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PikoAi
-Version: 0.1.18
+Version: 0.1.19
 Summary: An AI-powered task automation tool
 Home-page: https://github.com/nihaaaar22/OS-Assistant
 Author: Nihar S

{pikoai-0.1.18 → pikoai-0.1.19}/Src/llm_interface/llm.py

@@ -7,26 +7,11 @@ import os
 import sys
 import json
 import litellm # Added import for litellm
-import logging
-from datetime import datetime
 
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../../')))
 
 from Utils.ter_interface import TerminalInterface
 
-# Set up logging
-log_dir = os.path.join(os.path.dirname(__file__), '../../logs')
-os.makedirs(log_dir, exist_ok=True)
-log_file = os.path.join(log_dir, 'llm_responses.log')
-
-# Configure logging
-logging.basicConfig(
-    level=logging.INFO,
-    format='%(asctime)s - %(message)s',
-    handlers=[
-        logging.FileHandler(log_file)  # Only log to file, removed StreamHandler
-    ]
-)
 
 # Load environment variables from .env file
 load_dotenv()
@@ -45,15 +30,6 @@ class LiteLLMInterface:
     def __init__(self):
         self.terminal = TerminalInterface()
         self.model_name = self.load_config()
-        logging.info(f"\n{'='*50}\nNew Session - Using model: {self.model_name}\n{'='*50}")
-
-    def log_response(self, response_content):
-        """Log only the LLM response in a readable format."""
-        timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-        log_entry = f"\n{'='*50}\nTimestamp: {timestamp}\nModel: {self.model_name}\n\n"
-        log_entry += f"Response:\n{response_content}\n"
-        log_entry += f"{'='*50}\n"
-        logging.info(log_entry)
 
     def load_config(self):
         config_path = os.path.join(os.path.dirname(__file__), '../../config.json')
@@ -84,12 +60,10 @@ class LiteLLMInterface:
                     response_content += content
             
             self.terminal.flush_markdown()
-            # Log only the response after successful completion
-            self.log_response(response_content)
             return response_content
         except Exception as e:
-            # Log the error
-            logging.error(f"\n{'='*50}\nError occurred:\nModel: {self.model_name}\nError: {str(e)}\n{'='*50}")
+            # 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

{pikoai-0.1.18 → pikoai-0.1.19}/setup.py

@@ -3,7 +3,7 @@ from pathlib import Path
 
 setup(
     name="PikoAi",
-    version="0.1.18",
+    version="0.1.19",
     packages=find_packages(where="Src"),
     py_modules=["cli", "OpenCopilot"],
     package_dir={"": "Src"},