genarative cores
This commit is contained in:
@@ -0,0 +1,47 @@
|
||||
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"])
|
||||
logger.info(json_response["output"][0])
|
||||
logger.info(json_response["output"][0]["content"][0]["text"])
|
||||
# return json_response["generated_text"]
|
||||
|
||||
# chatgpt_response = response.choices[0].message.content
|
||||
return json_response
|
||||
except Exception as e:
|
||||
logger.error(f"ChatGpt Call Error {e}")
|
||||
return None
|
||||
Reference in New Issue
Block a user