diff options
Diffstat (limited to 'apps/docs/test.py')
| -rw-r--r-- | apps/docs/test.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/apps/docs/test.py b/apps/docs/test.py index ee5309d6..85ce6f82 100644 --- a/apps/docs/test.py +++ b/apps/docs/test.py @@ -12,14 +12,18 @@ conversation = [ # Get user profile + relevant memories for context profile = client.profile(container_tag=USER_ID, q=conversation[-1]["content"]) +static = "\n".join(profile.profile.static) +dynamic = "\n".join(profile.profile.dynamic) +memories = "\n".join(r.get("memory", "") for r in profile.search_results.results) + context = f"""Static profile: -{ "\n".join(profile.profile.static)} +{static} Dynamic profile: -{"\n".join(profile.profile.dynamic)} +{dynamic} Relevant memories: -{"\n".join(r.content for r in profile.search_results.results)}""" +{memories}""" # Build messages with memory-enriched context messages = [{"role": "system", "content": f"User context:\n{context}"}, *conversation] |