CHIEFSOFT\ameye
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
import stripe
|
|
import json
|
|
import logging
|
|
from app.config import settings, Config
|
|
# import openai
|
|
from flask import jsonify
|
|
from openai import OpenAI
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class GenerativeEngineIntegration:
|
|
OPENAI_API_KEY = Config.OPENAI_API_KEY
|
|
|
|
@staticmethod
|
|
def ask_generative(generative_prompt):
|
|
try:
|
|
# response = GenerativeEngine.openai.chat.completions.create(
|
|
# model="gpt-3.5-turbo", # Or other suitable model
|
|
# messages=[{"role": "user", "content": generative_prompt}]
|
|
# )
|
|
|
|
client = OpenAI(api_key=GenerativeEngineIntegration.OPENAI_API_KEY)
|
|
response = client.responses.create(
|
|
model="gpt-4.1",
|
|
input=generative_prompt
|
|
)
|
|
|
|
# input = "Tell me a three sentence bedtime story about a unicorn."
|
|
# logger.info(response)
|
|
json_response = json.loads(response.json())
|
|
# json_response = response.json()
|
|
# logger.info(json_response.keys())
|
|
# logger.info(json_response)
|
|
# logger.info(json_response["output"])
|
|
# logger.info(json_response["output"][0])
|
|
# logger.info(json_response["output"][0]["content"][0]["text"])
|
|
|
|
# chatgpt_response = response.choices[0].message.content
|
|
return json_response
|
|
except Exception as e:
|
|
logger.error(f"ChatGpt Call Error {e}")
|
|
return None
|