File size: 1,951 Bytes
47d267e 491704c 47d267e 491704c 47d267e 491704c 47d267e 491704c 47d267e 491704c 47d267e 491704c 47d267e 491704c 47d267e 491704c 47d267e 491704c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 |
import os
from dotenv import load_dotenv
from google import genai
from google.genai import types
from utils.prompts import icf_gemini_prompt
# Carrega as variáveis de ambiente (se você usar .env)
load_dotenv()
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
MODEL_ID = os.getenv("MODEL_ID")
CONTEXT_FIXED = ""
context_path = os.path.join(os.getcwd(), "RAG", "CIF_Lista.txt")
try:
with open(context_path, "r", encoding="utf-8") as f:
CONTEXT_FIXED = f.read()
except FileNotFoundError:
CONTEXT_FIXED = "Erro: Arquivo de contexto não encontrado."
print("Context: ", CONTEXT_FIXED[:100])
def api_generate(user_input: str) -> str:
client = genai.Client(api_key=GEMINI_API_KEY)
llm_config = types.GenerateContentConfig(
thinking_config=types.ThinkingConfig(
thinking_budget=-1,
),
response_mime_type="text/plain",
seed=1,
system_instruction=[
types.Part.from_text(text=icf_gemini_prompt),
],
)
user_prompt_content = types.Content(
role="user",
parts=[types.Part.from_text(text=CONTEXT_FIXED), types.Part.from_text(text=user_input)],
)
response = client.models.generate_content(model=MODEL_ID, contents=user_prompt_content, config=llm_config)
return response.text
if __name__ == "__main__":
test_string = "O paciente sente dores abdominais agudas, localizadas principalmente na região inferior do abdômen. Fadiga. Náuseas. Vômitos. Diarreia. Dificuldade para respirar. Dor no peito. O paciente observa vermelhidão persistente na pele, acompanhada de coceira em áreas específicas. Eu não consigo enxergar objetos a longas distâncias, com visão embaçada ao tentar focar. Tontura ou perda de equilíbrio. O paciente apresenta fraqueza súbita em um lado do corpo, dificultando movimentos do braço e perna."
print(f"Enviando...\n{test_string}")
res = api_generate(test_string)
print(res)
|