change boto3 to anthropic

app.py

@@ -5,7 +5,7 @@ import json
 import os
 from typing import List
 
-import boto3
+from anthropic import Anthropic
 import requests
 import streamlit as st
 from googleapiclient.discovery import build
@@ -13,7 +13,9 @@ from googleapiclient.discovery import build
 st.title("Qiitaに聞いた！！")
 
 if "client" not in st.session_state:
-    st.session_state.client = boto3.client("bedrock-runtime")
+    st.session_state.client = Anthropic(
+        api_key=os.environ.get("ANTHROPIC_API_KEY"),
+    )
 client = st.session_state.client
 
 
@@ -28,33 +30,25 @@ def generate_search_queries(question: str) -> List[str]:
         Format: {"queries": ["query_1", "query_2", "query_3"]}
         """
 
-    response = client.invoke_model(
-        modelId="anthropic.claude-3-haiku-20240307-v1:0",
-        body=json.dumps(
+    response = client.messages.create(
+        max_tokens=1024,
+        system="You are an expert at generating search queries for the Google search engine. Generate two search queries that are relevant to this question in Japanese. Output only valid JSON.",
+        messages=[
             {
-                "anthropic_version": "bedrock-2023-05-31",
-                "max_tokens": 1024,
-                "system": "You are an expert at generating search queries for the Google search engine. Generate two search queries that are relevant to this question in Japanese. Output only valid JSON.",
-                "messages": [
+                "role": "user",
+                "content": [
                     {
-                        "role": "user",
-                        "content": [
-                            {
-                                "type": "text",
-                                "text": GENERATE_QUERIES.replace(
-                                    "{{question}}", question
-                                ),
-                            }
-                        ],
-                    },
+                        "type": "text",
+                        "text": GENERATE_QUERIES.replace("{{question}}", question),
+                    }
                 ],
-                "temperature": 0,
             }
-        ),
+        ],
+        temperature=0,
+        model="claude-3-haiku-20240307",
     )
 
-    result = json.loads(response.get("body").read())
-    search_queries = result["content"][0]["text"]
+    search_queries = response.content[0].text
     search_queries = json.loads(search_queries)
 
     return search_queries
@@ -133,30 +127,24 @@ def generate_answer(question: str, documents: dict):
     User's question: {question}
     """
 
-    response = client.invoke_model(
-        modelId="anthropic.claude-3-haiku-20240307-v1:0",
-        body=json.dumps(
+    response = client.messages.create(
+        max_tokens=1024,
+        messages=[
             {
-                "anthropic_version": "bedrock-2023-05-31",
-                "max_tokens": 1024,
-                "messages": [
+                "role": "user",
+                "content": [
                     {
-                        "role": "user",
-                        "content": [
-                            {
-                                "type": "text",
-                                "text": ANSWER_QUESTION,
-                            }
-                        ],
-                    },
+                        "type": "text",
+                        "text": ANSWER_QUESTION,
+                    }
                 ],
-                "temperature": 0.1,
             }
-        ),
+        ],
+        temperature=0.1,
+        model="claude-3-haiku-20240307",
     )
 
-    result = json.loads(response.get("body").read())
-    return result["content"][0]["text"]
+    return response.content[0].text
 
 
 # メイン関数

requirements.txt

@@ -1,3 +1,3 @@
-boto3
+anthropic
 streamlit
 google-api-python-client