RAGを開発してみたい！ので簡単なものでもいいので作ってみよう！part6

Sat, 04 May 2024 00:00:00 +0000

PCが故障したという話をしたのですが、コンセントから電源が抜けてたというオチでした。

なのでRAGの開発を再開しました！

前回開発した部分は

質疑応答画面の作成

今回開発する部分は

画面(Next.js)から送られるデータを受け取るエンドポイントをFlaskに設定
ベクトル化するプログラムをクラス化

Flaskも触ったことないのでChat-GPTに質問しながら触ってました。一旦中身は以下の内容にしています。

from flask import Flask, request, jsonify
from flask_cors import CORS
import pdf_to_model
import logging
logging.basicConfig(level=logging.DEBUG)
app = Flask(__name__)
CORS(app)
@app.route('/process', methods=['POST'])
def process_question():
data = request.get_json()
query = data.get('query')
logging.debug("Received query: %s", query)
try:
vector = pdf_to_model.PDFToVector()
result = vector.main(query)
# 結果を JSON 形式で返す
return jsonify(result[0])
# return jsonify({'result': 'Processed result of ' + query})
except Exception as e:
logging.error("Error processing query: %s", str(e))
return jsonify({'error': str(e)}), 500
if __name__ == "__main__":
app.run(debug=True)

ログの設定はする必要はないですが、まだ開発段階なので一旦設定しています。それから"CORS(app)“ですね。少し調べたのですが異なるオリジン間でのリソース共有を可能にするために必要な設定みたいです。そもそも同一のオリジンとは以下の3つが同じであることが条件です。

Corsapp on Xainome Blog

RAGを開発してみたい！ので簡単なものでもいいので作ってみよう！part6