From 71013c381f63c42ecc95f772e5b2437ceeaec5a0 Mon Sep 17 00:00:00 2001
From: Mark Ceter <133643956+maceter@users.noreply.github.com>
Date: Wed, 17 May 2023 18:37:22 +0000
Subject: [PATCH] Add arbitrary metadata to message

---
 README.md | 6 ++++--
 server.py | 3 ++-
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 2bdbafb..b33fd13 100644
--- a/README.md
+++ b/README.md
@@ -333,7 +333,8 @@ WAV audio file.
             "id": "633a4bd1-8350-46b5-9ef2-f5d27acdecb7", 
             "date": 1684164339877,
             "role": "user",
-            "content": "Hello, AI world!"
+            "content": "Hello, AI world!",
+            "meta": "this is meta"
         },
         {
             "id": "8a2ed36b-c212-4a1b-84a3-0ffbe0896506", 
@@ -367,7 +368,8 @@ WAV audio file.
         "date": 1684164339877,
         "role": "user",
         "content": "Hello, AI world!",
-        "distance": 0.31
+        "distance": 0.31,
+        "meta": "this is meta"
     },
     {
         "id": "8a2ed36b-c212-4a1b-84a3-0ffbe0896506", 
diff --git a/server.py b/server.py
index 44a009e..d60b627 100644
--- a/server.py
+++ b/server.py
@@ -640,7 +640,7 @@ def chromadb_add_messages():
 
     documents = [m['content'] for m in data['messages']]
     ids = [m['id'] for m in data['messages']]
-    metadatas = [{'role': m['role'], 'date': m['date']} for m in data['messages']]
+    metadatas = [{'role': m['role'], 'date': m['date'], 'meta': m.get('meta', '')} for m in data['messages']]
 
     collection.upsert(
         ids=ids,
@@ -687,6 +687,7 @@ def chromadb_query():
             'id': ids[i],
             'date': metadatas[i]['date'],
             'role': metadatas[i]['role'],
+            'meta': metadatas[i]['meta'],
             'content': documents[i],
             'distance': distances[i]
         } for i in range(len(ids))