From ae39b12a70fb67d7cd753567b8c644a8ac2f3731 Mon Sep 17 00:00:00 2001
From: birbbit <hristijan.bogoevski@gmail.com>
Date: Sun, 11 Feb 2024 02:53:23 -0800
Subject: [PATCH] Skills added dynamically to prompt

---
 .gitignore               |  3 +++
 OS/01/i.py               | 37 ++++++++++++++++++++++++++++++-------
 OS/01/skills/schedule.py |  8 +++++---
 3 files changed, 38 insertions(+), 10 deletions(-)

diff --git a/.gitignore b/.gitignore
index 0abf9bb..3544e4d 100644
--- a/.gitignore
+++ b/.gitignore
@@ -161,3 +161,6 @@ cython_debug/
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
 
+
+# ignore the aifs index files
+_.aifs
diff --git a/OS/01/i.py b/OS/01/i.py
index 0c24c49..805bf86 100644
--- a/OS/01/i.py
+++ b/OS/01/i.py
@@ -5,7 +5,10 @@ import os
 import glob
 import json
 from pathlib import Path
+from interpreter import OpenInterpreter
 
+
+def configure_interpreter(interpreter: OpenInterpreter):
 def configure_interpreter(interpreter):
     ### SYSTEM MESSAGE
 
@@ -41,6 +44,31 @@ def configure_interpreter(interpreter):
 
     Remember: You can run Python code. Be very concise. Ensure that you actually run code every time! THIS IS IMPORTANT. You NEED to write code. **Help the user by being very concise in your answers.** Do not break down tasks excessively, just into simple, few minute steps. Don't assume the user lives their life in a certain way— pick very general tasks if you're breaking a task down.
 
+    Use the following functions (assume they're imported) to complete your goals whenever possible:
+    {{
+import sys
+
+original_stdout = sys.stdout
+sys.stdout = open(os.devnull, 'w')
+original_stderr = sys.stderr
+sys.stderr = open(os.devnull, 'w')
+
+from interpreter import interpreter
+from pathlib import Path
+
+query = "all functions"
+skills_path = Path().resolve() / 'skills'
+paths_in_skills = [str(path) for path in skills_path.glob('**/*.py')]
+skills = interpreter.computer.docs.search(query, paths=paths_in_skills)
+lowercase_skills = [skill[0].lower() + skill[1:] for skill in skills]
+output = "\\n".join(lowercase_skills)
+
+sys.stdout = original_stdout
+sys.stderr = original_stderr
+
+print(output)
+    }}
+
     """.strip()
 
     interpreter.custom_instructions = system_message
@@ -65,7 +93,6 @@ def configure_interpreter(interpreter):
     interpreter.offline = True
     interpreter.id = 206 # Used to identify itself to other interpreters. This should be changed programatically so it's unique.
 
-
     ### RESET conversations/user.json
 
     script_dir = os.path.dirname(os.path.abspath(__file__))
@@ -74,11 +101,7 @@ def configure_interpreter(interpreter):
         json.dump([], file)
 
     ### SKILLS
-
-    skills_path = Path(__file__).parent / 'skills'
-    for file in glob.glob(os.path.join(skills_path, '*.py')):
-        with open(file, 'r') as f:
-            for chunk in interpreter.computer.run("python", f.read()):
-                print(chunk)
+    interpreter.computer.skills.skills_dir = Path(__file__).parent / 'skills'
+    interpreter.computer.skills.import_skills()
 
     return interpreter
\ No newline at end of file
diff --git a/OS/01/skills/schedule.py b/OS/01/skills/schedule.py
index fc62147..2292d42 100644
--- a/OS/01/skills/schedule.py
+++ b/OS/01/skills/schedule.py
@@ -3,8 +3,8 @@ from datetime import datetime
 import json
 import subprocess
 
-def add_message_to_queue(message):
 
+def _add_message_to_queue(message):
     # Define the message data and convert it to JSON
     message_json = json.dumps({
         "role": "computer",
@@ -14,12 +14,14 @@ def add_message_to_queue(message):
     })
     subprocess.run(['logger', '{TO_INTERPRETER{' + message_json + '}TO_INTERPRETER}'])
 
-def schedule(dt, message):
+
+def schedule(dt: datetime, message: str) -> None:
+    """"Schedules a reminder at a specific time. At the specified time, the message will be added to the queue."""
     # Calculate the delay in seconds
     delay = (dt - datetime.now()).total_seconds()
 
     # Create a timer
-    timer = threading.Timer(delay, add_message_to_queue, args=[message])
+    timer = threading.Timer(delay, _add_message_to_queue, args=[message])
 
     # Start the timer
     timer.start()
\ No newline at end of file