nginx-template / api.py
deepak191z's picture
Update api.py
ef5f149 verified
raw
history blame
1.45 kB
from flask import Flask, request, jsonify
from duckai import DuckAI
from flask_cors import CORS
app = Flask(__name__)
CORS(app, resources={r"/*": {
"origins": "*",
"allow_headers": "*",
"methods": ["GET", "POST", "PUT", "DELETE", "OPTIONS"],
"supports_credentials": True
}})
@app.route("/chat/", methods=["GET"])
def chat():
query = request.args.get("query")
if not query:
return jsonify({"error": "Query parameter is required"}), 400
duck = DuckAI() # create one DuckAI instance
try:
results = duck.chat(query, model='gpt-4o-mini')
return jsonify({"results": results})
except Exception as e1:
print(f"Primary model (gpt-4o-mini) failed: {e1}")
try:
results = duck.chat(query, model='claude-3-haiku')
return jsonify({"results": results})
except Exception as e2:
print(f"Fallback model (claude-3-haiku) also failed: {e2}")
return jsonify({
"error": "Both models failed",
"primary_error": str(e1),
"fallback_error": str(e2)
}), 500
def chat_with_model(query: str, model: str):
try:
duck = DuckAI()
results = duck.chat(query, model=model)
return jsonify({"results": results})
except Exception as e:
return jsonify({"error": str(e)}), 500
if __name__ == "__main__":
app.run(host="localhost", port=3000, debug=True)