From 70557afb1e09fc99e9488b69322a76c50dc98554 Mon Sep 17 00:00:00 2001 From: CaptainJack2491 Date: Sun, 16 Nov 2025 19:29:30 +0000 Subject: added modular prompts --- src/agents/03-logs/agent.py | 10 +++++++--- src/agents/03-logs/main.py | 10 ++++++++-- src/agents/03-logs/prompts/default/system.md | 1 + src/agents/03-logs/prompts/default/user.md | 1 + 4 files changed, 17 insertions(+), 5 deletions(-) create mode 100644 src/agents/03-logs/prompts/default/system.md create mode 100644 src/agents/03-logs/prompts/default/user.md 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 -- cgit v1.2.3