chore: migrate package name to letta (#1775)

Co-authored-by: Charles Packer <packercharles@gmail.com> Co-authored-by: Shubham Naik <shubham.naik10@gmail.com> Co-authored-by: Shubham Naik <shub@memgpt.ai>
2024-09-23 09:15:18 -07:00
parent 9ebbaacc1f
commit 8ae1e64987
337 changed files with 5528 additions and 6795 deletions
--- a/letta/client/utils.py
+++ b/letta/client/utils.py
@@ -0,0 +1,61 @@
+from datetime import datetime
+
+from IPython.display import HTML, display
+
+
+def pprint(messages):
+    """Utility function for pretty-printing the output of client.send_message in notebooks"""
+
+    css_styles = """
+    <style>
+        .terminal {
+            background-color: #002b36;
+            color: #839496;
+            font-family: 'Courier New', Courier, monospace;
+            padding: 10px;
+            border-radius: 5px;
+        }
+        .terminal strong {
+            color: #b58900;
+        }
+        .terminal .function-return {
+            color: #2aa198;
+        }
+        .terminal .internal-monologue {
+            color: #d33682;
+        }
+        .terminal .function-call {
+            color: #2aa198;
+        }
+        .terminal .assistant-message {
+            color: #859900;
+        }
+        .terminal pre {
+            color: #839496;
+        }
+    </style>
+    """
+
+    html_content = css_styles + "<div class='terminal'>"
+    for message in messages:
+        date_str = message["date"]
+        date_formatted = datetime.fromisoformat(date_str.replace("Z", "+00:00")).strftime("%Y-%m-%d %H:%M:%S")
+
+        if "function_return" in message:
+            return_string = message["function_return"]
+            return_status = message["status"]
+            html_content += f"<p><strong>🛠️ [{date_formatted}] Function Return ({return_status}):</strong></p>"
+            html_content += f"<p class='function-return'>{return_string}</p>"
+        elif "internal_monologue" in message:
+            html_content += f"<p><strong>💭 [{date_formatted}] Internal Monologue:</strong></p>"
+            html_content += f"<p class='internal-monologue'>{message['internal_monologue']}</p>"
+        elif "function_call" in message:
+            html_content += f"<p><strong>🛠️ [[{date_formatted}] Function Call:</strong></p>"
+            html_content += f"<p class='function-call'>{message['function_call']}</p>"
+        elif "assistant_message" in message:
+            html_content += f"<p><strong>🤖 [{date_formatted}] Assistant Message:</strong></p>"
+            html_content += f"<p class='assistant-message'>{message['assistant_message']}</p>"
+        html_content += "<br>"
+    html_content += "</div>"
+
+    display(HTML(html_content))