PikoAi 0.1.18__py3-none-any.whl → 0.1.19__py3-none-any.whl

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.dist-info → pikoai-0.1.19.dist-info}/METADATA

@@ -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.dist-info → pikoai-0.1.19.dist-info}/RECORD

@@ -26,10 +26,10 @@ Utils/__init__.py,sha256=oukU0ufroPRd8_N8d2xiFes9CTxSaw4NA6p2nS1kkSg,16
 Utils/executor_utils.py,sha256=WwK3TKgw_hG_crg7ijRaqfidYnnNXYbbs37vKZRYK-0,491
 Utils/ter_interface.py,sha256=8Oe5818MAYC21SfUxtfnV9HQFcQ49z8Q030jjPqNP_g,3889
 llm_interface/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-llm_interface/llm.py,sha256=TEKKgRfZGtk_UgBfqqkdWKHr4NS3jOOk9di0S3RA7c0,6216
-pikoai-0.1.18.dist-info/licenses/LICENSE,sha256=cELUVOboOAderKFp8bdtcM5VyJi61YH1oDbRhOuoQZw,1067
-pikoai-0.1.18.dist-info/METADATA,sha256=BM1IQxmZ4D7p07EfgLAuUGhpLvAT3W9oPw9WTV8Lee0,2962
-pikoai-0.1.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pikoai-0.1.18.dist-info/entry_points.txt,sha256=xjZnheDymNDnQ0o84R0jZKEITrhNbzQWN-AhqfA_d6s,50
-pikoai-0.1.18.dist-info/top_level.txt,sha256=hWzBNE7UQsuNcENIOksGcJED08k3ZGRRn2X5jnStICU,53
-pikoai-0.1.18.dist-info/RECORD,,
+llm_interface/llm.py,sha256=tI_KDOW14QLWowA7bB3GPe2qjlk0sjS5fBavs9XD1fo,5185
+pikoai-0.1.19.dist-info/licenses/LICENSE,sha256=cELUVOboOAderKFp8bdtcM5VyJi61YH1oDbRhOuoQZw,1067
+pikoai-0.1.19.dist-info/METADATA,sha256=4N9c6Htp6lsrimD8ukdfxTZG59quiJodw6pIECoLVPg,2962
+pikoai-0.1.19.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pikoai-0.1.19.dist-info/entry_points.txt,sha256=xjZnheDymNDnQ0o84R0jZKEITrhNbzQWN-AhqfA_d6s,50
+pikoai-0.1.19.dist-info/top_level.txt,sha256=hWzBNE7UQsuNcENIOksGcJED08k3ZGRRn2X5jnStICU,53
+pikoai-0.1.19.dist-info/RECORD,,