Update app.py

app.py

@@ -1,35 +1,71 @@
-from os import environ
-from utils.ai import (
-    retrieve_context,
-    construct_prompt,
-    get_remote_chat_response,
-)
+from context import query
+from openai import AsyncOpenAI
 import panel as pn
 
+# taken from fleet context
+SYSTEM_PROMPT = """
+You are an expert in Python libraries. You carefully provide accurate, factual, thoughtful, nuanced answers, and are brilliant at reasoning. If you think there might not be a correct answer, you say so.
+Each token you produce is another opportunity to use computation, therefore you always spend a few sentences explaining background context, assumptions, and step-by-step thinking BEFORE you try to answer a question.
+Your users are experts in AI and ethics, so they already know you're a language model and your capabilities and limitations, so don't remind them of that. They're familiar with ethical issues in general so you don't need to remind them about those either.
+Your users are also in a CLI environment. You are capable of writing and running code. DO NOT write hypothetical code. ALWAYS write real code that will execute and run end-to-end.
+
+Instructions:
+- Be objective, direct. Include literal information from the context, don't add any conclusion or subjective information.
+- When writing code, ALWAYS have some sort of output (like a print statement). If you're writing a function, call it at the end. Do not generate the output, because the user can run it themselves.
+- ALWAYS cite your sources. Context will be given to you after the text ### Context source_url ### with source_url being the url to the file. For example, ### Context https://example.com/docs/api.html#files ### will have a source_url of https://example.com/docs/api.html#files.
+- When you cite your source, please cite it as [num] with `num` starting at 1 and incrementing with each source cited (1, 2, 3, ...). At the bottom, have a newline-separated `num: source_url` at the end of the response. ALWAYS add a new line between sources or else the user won't be able to read it. DO NOT convert links into markdown, EVER! If you do that, the user will not be able to click on the links.
+For example:
+**Context 1**: https://example.com/docs/api.html#pdfs
+I'm a big fan of PDFs.
+**Context 2**: https://example.com/docs/api.html#csvs
+I'm a big fan of CSVs.
+### Prompt ###
+What is this person a big fan of?
+### Response ###
+This person is a big fan of PDFs[1] and CSVs[2].
+1: https://example.com/docs/api.html#pdfs
+2: https://example.com/docs/api.html#csvs
+"""
+
 pn.extension()
 
 MODEL = "gpt-3.5-turbo"
 
 
-def seek(contents, user, instance):
-    messages = instance.serialize()[1:-1]
-    rag_context = retrieve_context(contents, k=1)
-    prompts = construct_prompt(
-        messages,
-        rag_context,
-        model=MODEL,
-        cite_sources=True,
-        context_window=4097,
+async def answer(contents, user, instance):
+    # start with system prompt
+    messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+
+    # add context to the user input
+    context = ""
+    fleet_responses = query(contents, k=3)
+    for i, response in enumerate(fleet_responses):
+        context += f"\n\n**Context {i}**: {response['metadata']['url']}\n{response['metadata']['text']}"
+    instance.send(context, avatar="🛩️", user="Fleet Context", respond=False)
+
+    # get history of messages (skipping the intro message)
+    # and serialize fleet context messages as "user" role
+    messages.extend(
+        instance.serialize(role_names={"user": ["user", "Fleet Context"]})[1:]
+    )
+
+    openai_response = await client.chat.completions.create(
+        model=MODEL, messages=messages, temperature=0.2, stream=True
     )
 
-    message = None
-    for response in get_remote_chat_response(prompts, model=MODEL):
-        if response:
-            message = instance.stream(response, avatar="🛩️", user="Fleet Context", message=message)
+    message = ""
+    async for chunk in openai_response:
+        token = chunk.choices[0].delta.content
+        if token:
+            message += token
+            yield message
 
 
-chat_interface = pn.chat.ChatInterface(callback=seek, callback_exception="verbose")
-chat_interface.send("Ask me anything about Python libraries!", avatar="🛩️", user="Fleet Context", respond=False)
+client = AsyncOpenAI()
+chat_interface = pn.chat.ChatInterface(callback=answer, callback_user="OpenAI", callback_exception="verbose")
+chat_interface.send(
+    "Ask me anything about Python libraries!", user="System", respond=False,
+)
 
-template = pn.template.FastListTemplate(main=[chat_interface], title="Panel UI of Fleet Context")
+template = pn.template.FastListTemplate(main=[chat_interface], title="Panel UI of Fleet Context 🛩️")
 template.servable()