added modular prompts

4 files changed, 17 insertions, 5 deletions

diff --git a/src/agents/03-logs/agent.py b/src/agents/03-logs/agent.py
index 1f19431..657e41a 100644
--- a/src/agents/03-logs/agent.py
+++ b/src/agents/03-logs/agent.py
@@ -6,17 +6,21 @@ from config import MODEL, BASE_URL, API_KEY, TEMPERATURE
 from tools import tools, available_functions
 
 class Agent:
-    def __init__(self, model=MODEL, base_url=BASE_URL, api_key=API_KEY, temperature=TEMPERATURE):
+    def __init__(self, system_prompt="You are a helpful assistant.", model=MODEL, base_url=BASE_URL, api_key=API_KEY, temperature=TEMPERATURE):
         self.client = OpenAI(base_url=base_url, api_key=api_key)
         self.model = model
         self.temperature = temperature
         self.tools = tools
         self.available_functions = available_functions
         self.logs = []
+        self.system_prompt = system_prompt
 
     def run(self, initial_prompt):
-        messages = [{'role': 'user', 'content': initial_prompt}]
-        self.logs.append(messages[0])
+        messages = [
+            {'role': 'system', 'content': self.system_prompt},
+            {'role': 'user', 'content': initial_prompt}
+        ]
+        self.logs.extend(messages)
         
         while True:
             response = self.client.chat.completions.create(
diff --git a/src/agents/03-logs/main.py b/src/agents/03-logs/main.py
index d224048..e9f0b01 100644
--- a/src/agents/03-logs/main.py
+++ b/src/agents/03-logs/main.py
@@ -1,8 +1,14 @@
 # main.py
 from agent import Agent
 
+def load_prompt(file_path):
+    with open(file_path, 'r') as f:
+        return f.read()
+
 if __name__ == "__main__":
-    user_prompt = "check if this username is valid 'jack'."
-    agent = Agent()
+    system_prompt = load_prompt('prompts/default/system.md')
+    user_prompt = load_prompt('prompts/default/user.md')
+    
+    agent = Agent(system_prompt=system_prompt)
     agent.run(user_prompt)
     agent.save_logs()
\ No newline at end of file
diff --git a/src/agents/03-logs/prompts/default/system.md b/src/agents/03-logs/prompts/default/system.md
new file mode 100644
index 0000000..ac39ad9
--- /dev/null
+++ b/src/agents/03-logs/prompts/default/system.md
@@ -0,0 +1 @@
+You are a helpful assistant.
\ No newline at end of file
diff --git a/src/agents/03-logs/prompts/default/user.md b/src/agents/03-logs/prompts/default/user.md
new file mode 100644
index 0000000..7a075e3
--- /dev/null
+++ b/src/agents/03-logs/prompts/default/user.md
@@ -0,0 +1 @@
+check if this username is valid 'jack'.
\ No newline at end of file