fix llm app examples

examples/streamlit/llm_app.py

@@ -23,7 +23,7 @@
 else:
     st.warning("To save some feedback to Trubrics, add your account details in the sidebar.")
 
-models = ("text-davinci-003", "text-davinci-002")
+models = ("gpt-3.5-turbo",)
 model = st.selectbox(
     "Choose your GPT-3.5 LLM",
     models,
@@ -38,10 +38,10 @@
 button = st.button(f"Ask {model}")
 
 if button:
-    response = openai.Completion.create(model=model, prompt=prompt, temperature=0.5, max_tokens=200)
-    response_text = response.choices[0].text.replace("\n", "")
+    response = openai.ChatCompletion.create(model=model, messages=[{"role": "user", "content": prompt}])
+    response_text = response.choices[0].message["content"]
     st.session_state.logged_prompt = collector.log_prompt(
-        model_config={"model": model}, prompt=prompt, generation=response_text
+        model_config={"model": model}, prompt=prompt, generation=response_text, tags=["llm_app.py"]
     )
     st.session_state.response = response_text
 
@@ -56,6 +56,7 @@
         model=model,
         align="flex-start",
         single_submit=False,
+        tags=["llm_app.py"],
     )
 
     if feedback:

examples/streamlit/llm_chatbot.py

@@ -27,7 +27,7 @@ def init_trubrics(email, password):
 if "session_id" not in st.session_state:
     st.session_state["session_id"] = str(uuid.uuid4())
 
-model = "gpt-4"
+model = "gpt-3.5-turbo"
 
 openai_api_key = st.secrets.get("OPENAI_API_KEY")
 if prompt := st.chat_input():
@@ -41,7 +41,11 @@ def init_trubrics(email, password):
     response = openai.ChatCompletion.create(model=model, messages=st.session_state.messages)
     msg = response.choices[0].message
     logged_prompt = collector.log_prompt(
-        model_config={"model": model}, prompt=prompt, generation=msg["content"], session_id=st.session_state.session_id
+        model_config={"model": model},
+        prompt=prompt,
+        generation=msg["content"],
+        session_id=st.session_state.session_id,
+        tags=["llm_chatbot.py"],
     )
     st.session_state.prompt_ids.append(logged_prompt.id)
     st.session_state.messages.append(msg)
@@ -61,6 +65,7 @@ def init_trubrics(email, password):
                 align="flex-end",
                 single_submit=True,
                 key=f"feedback_{int(n/2)}",
+                tags=["llm_chatbot.py"],
             )
 
         else:

requirements.txt

@@ -1,4 +1,4 @@
 requests>=2.28.0
-pydantic>=1.9.1,<2.0
+pydantic>=1.9.1
 typer>=0.6.1
 loguru>=0.6.0