import os
from memoryos import Memory
from openai import OpenAI
def main() -> None:
memory = Memory(api_key=os.environ["MEMORYOS_API_KEY"])
openai_client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
external_user_id = "customer-123"
user_message = "What kind of answer format do I like?"
memories = memory.get(
query=user_message,
external_user_id=external_user_id,
limit=5,
)
system_prompt = "You are a helpful assistant."
if memories.has_context:
system_prompt = f"{system_prompt}\n\n{memories.system_prompt_addition}"
response = openai_client.responses.create(
model="gpt-4.1-mini",
input=[
{"role": "system", "content": system_prompt},
{"role": "user", "content": user_message},
],
)
print(response.output_text)
memory.close()
if __name__ == "__main__":
main()