feat: server

src/gpt.py

@@ -2,9 +2,13 @@ import os
 
 import openai
 
+from src.utils.string import print_colored
+
 openai.api_key = os.environ['OPENAI_API_KEY']
 
 def get_response(system_definition, user_query):
+    print_colored('system_definition', system_definition, 'magenta')
+    print_colored('user_query', user_query, 'blue')
     response = openai.ChatCompletion.create(
         temperature=0,
         model="gpt-4",
@@ -23,5 +27,5 @@ def get_response(system_definition, user_query):
         ]
     )
     content = response['choices'][0]['message']['content']
-    print(content)
+    print_colored('agent response', content, 'green')
     return content