35 lines
834 B
Python
35 lines
834 B
Python
from flask import Flask, jsonify, request
|
|
from dotenv import load_dotenv
|
|
from services.embedding_service import Embedding
|
|
|
|
|
|
load_dotenv()
|
|
app = Flask(__name__)
|
|
|
|
# @app.route("/")
|
|
# def hello():
|
|
# return jsonify({"message":"Hello, World!"})
|
|
|
|
@app.get("/")
|
|
def hello():
|
|
return jsonify({"message":"Hello, World!"})
|
|
|
|
|
|
@app.post("/get_embedding")
|
|
def get_embedding():
|
|
# Get the query text from request JSON
|
|
data = request.get_json()
|
|
query = data.get("query", "")
|
|
|
|
if not query:
|
|
return jsonify({"error": "Query text is required"}), 400
|
|
|
|
# Call the embedding function
|
|
vector = Embedding.call(query, is_query=True)
|
|
|
|
return jsonify({"query": query, "number of embedding": len(vector), "one embedding":len(vector[0])})
|
|
|
|
print(__name__)
|
|
|
|
if __name__=="__main__":
|
|
app.run(debug=True, port=8000) |