Add support for groq

.env.example

@@ -1,3 +1,4 @@
 OPENAI_API_KEY=
 ANTHROPIC_API_KEY=
 GOOGLE_API_KEY=
+GROQ_API_KEY=

app.py

@@ -1,17 +1,15 @@
 import os
 
+from groq import Groq
 from openai import OpenAI
+from anthropic import Anthropic
 import google.generativeai as genai
-from anthropic import Anthropic, HUMAN_PROMPT, AI_PROMPT
-import streamlit as st
+
 from dotenv import load_dotenv
+import streamlit as st
 
 load_dotenv()
 
-anthropic = Anthropic(api_key=os.environ.get("ANTHROPIC_API_KEY"))
-genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
-client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
-
 st.set_page_config(page_title="strftime AI", page_icon="🕘")
 
 with open("static/style.css") as f:
@@ -21,8 +19,14 @@ st.header("strftime AI")
 
 with st.sidebar:
     provider = st.radio(
-        "Choose AI Provider", options=["Anthropic", "Google Gemini", "OpenAI"]
-    )
+        "Choose AI Provider",
+        options=[
+            "Groq",
+            "Anthropic",
+            "OpenAI",
+            "Google Gemini",
+            ]
+        )
 
 text = st.text_input("Enter datetime text eg. 2023-09-28T15:27:58Z", value="2023-09-28T15:27:58Z")
 
@@ -32,20 +36,32 @@ if text:
         "show the strftime string on top and then breakdown of % symbols: " + text
     )
 
-    if provider == "Anthropic":
-        completion = anthropic.completions.create(
-            model="claude-2",
-            max_tokens_to_sample=300,
-            prompt=f"{HUMAN_PROMPT} {prompt}{AI_PROMPT}",
+    if provider == "Groq":
+        client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
+        chat_completion = client.chat.completions.create(
+            messages=[{"role": "user", "content": prompt}],
+            model="mixtral-8x7b-32768",
+            max_tokens=1024,
         )
-        st.markdown(completion.completion)
-    elif provider == "Google Gemini":
-        model = genai.GenerativeModel("gemini-pro")
-        response = model.generate_content(prompt)
-        st.markdown(response.text)
+        st.markdown(chat_completion.choices[0].message.content)
     elif provider == "OpenAI":
+        client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
         chat_completion = client.chat.completions.create(
+            messages=[{"role": "user", "content": prompt}],
             model="gpt-3.5-turbo",
-            messages=[{"role": "user", "content": prompt}]
+            max_tokens=1024,
         )
         st.markdown(chat_completion.choices[0].message.content)
+    elif provider == "Anthropic":
+        client = Anthropic(api_key=os.getenv("ANTHROPIC_API_KEY"))
+        message = client.messages.create(
+            messages=[{"role": "user", "content": prompt}],
+            model="claude-3-haiku-20240307",
+            max_tokens=1024,
+        )
+        st.markdown(message.content[0].text)
+    elif provider == "Google Gemini":
+        genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+        model = genai.GenerativeModel("gemini-pro")
+        response = model.generate_content(prompt)
+        st.markdown(response.text)

requirements.txt

@@ -1,5 +1,7 @@
+groq
 openai
 anthropic
 google-generativeai
+
 python-dotenv
 streamlit