Final_Assignment_Template / constantes.py
gdms's picture
Ajuste Chess + execução tabela comutativa
d272ca0
raw
history blame
1.49 kB
import os
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
TAVILY_API_KEY = os.getenv("TAVILY_API_KEY")
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
LANGCHAIN_API_KEY = os.getenv("LANGCHAIN_API_KEY")
os.environ["LANGCHAIN_TRACING_V2"] = "true"
os.environ["LANGCHAIN_PROJECT"] = "hf-final-assessment"
OUTPUT_AUDIO_PATH ="./tmp/audio_analysis_output"
AUDIO_FILENAME = "downloaded_audio"
TRANSCRIPT_FILENAME = "transcript.txt"
OUTPUT_IMAGE_PATH = "./tmp/image_analysis_output"
IMAGE_FILE_NAME = "downloaded_image"
OUTPUT_VIDEO_PATH = "./tmp/video_analysis_output"
VIDEO_FILE_NAME = "downloaded_video"
FRAME_INTERVAL_SECONDS = 0.5 # Intervalo entre frames a serem extraídos
INICIO_FRAME_IMPORTANTE = 191 # inicio intervalo relevante, para não ficar caro a inferencia ao gpt
FIM_FRAME_IMPORTANTE = 193# fim intervalo relevante, para não ficar caro a inferencia ao gpt
YOUTUBE_COOKIE_PATH = "./support/cookie-youtube.txt"
GPT_IMAGE_MODEL = "gpt-4o"
GEMINI_MODEL = "gemini-2.0-flash"
CHESSVISION_TO_FEN_URL = "http://app.chessvision.ai/predict"
CHESS_MOVE_API = "https://chess-api.com/v1"
AGENTS_FILES_PATH = "./tmp"
HUGGINGFACE_DATASET_URL_TEMPLATE = (
"https://huggingface.co/datasets/gdms/gaia/resolve/main/{filename}"
)
LISTA_TASKS_PROCESSAR = [
# "8e867cd7-cff9-4e6c-867a-ff5ddc2550be",
# "a1e91b78-d3d8-4675-bb8d-62741b4b68a6",
# "2d83110e-a098-4ebb-9987-066c06fa42d0",
"cca530fc-4052-43b2-b130-b30968d8aa44",
"6f37996b-2ac7-44b0-8e68-6d28256631b4"
]