Skip to content

Commit

Permalink
chatgpt assistant
Browse files Browse the repository at this point in the history
  • Loading branch information
harudra committed Aug 11, 2024
1 parent ea1aa39 commit e3e35fd
Showing 1 changed file with 13 additions and 2 deletions.
15 changes: 13 additions & 2 deletions api/services/gpt.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
from openai import OpenAI
from utils.config import get_openai_api_key
import os

OpenAI.api_key = get_openai_api_key()
client = OpenAI()
Expand All @@ -8,15 +9,25 @@ def generate_text(prompt: str) -> str:
assistant = client.beta.assistants.create(
name="futarin",
instructions=
"あなたは、プロのカウンセラーです。与えられたファイルをもとに相談者の気持ちに寄り添い、解決策を提示しながら、簡潔でフレンドリーなメッセージを生成してください。",
"あなたは、プロのカウンセラーです。与えられたファイルをもとに相談者の気持ちに寄り添い、解決策を提示しながら、簡潔でフレンドリーなひとまとまりのメッセージを生成してください。",
model="gpt-4o-mini",
tools=[{"type": "file_search"}],
)

vs_name = "professional counselor"
vector_store = client.beta.vector_stores.create(name=vs_name)

file_streams = [open("../utils/cheering.txt", "rb")]
# このスクリプトが存在するディレクトリのパスを取得
current_dir = os.path.dirname(os.path.abspath(__file__))

# cheering.txt への相対パスを作成
file_path = os.path.join(current_dir, '../utils/cheering.txt')

# ファイルを開く
# with open(file_path, 'r') as file:
# file_streams = file.read()

file_streams = [open(file_path, "rb")]

file_batch = client.beta.vector_stores.file_batches.upload_and_poll(
vector_store_id=vector_store.id, files=file_streams
Expand Down

0 comments on commit e3e35fd

Please sign in to comment.