Spaces:
Sleeping
Sleeping
fredcaixeta commited on
Commit ·
78e76ff
1
Parent(s): d043f81
goo
Browse files
agent.py
DELETED
|
@@ -1,43 +0,0 @@
|
|
| 1 |
-
import os
|
| 2 |
-
from dotenv import load_dotenv
|
| 3 |
-
|
| 4 |
-
from pydantic_ai import Agent, RunContext
|
| 5 |
-
# from pydantic_ai.messages import (ModelMessage, ModelRequest, ModelResponse, UserPromptPart,
|
| 6 |
-
# SystemPromptPart, TextPart, ModelMessagesTypeAdapter,
|
| 7 |
-
# ToolReturnPart)
|
| 8 |
-
from pydantic_ai.usage import UsageLimits
|
| 9 |
-
from pydantic_core import to_jsonable_python
|
| 10 |
-
from pydantic_ai.models.groq import GroqModel
|
| 11 |
-
|
| 12 |
-
from dataclasses import dataclass
|
| 13 |
-
|
| 14 |
-
from dotenv import load_dotenv
|
| 15 |
-
load_dotenv()
|
| 16 |
-
|
| 17 |
-
# Defina um prompt de sistema padrão
|
| 18 |
-
DEFAULT_SYSTEM_PROMPT = f"""
|
| 19 |
-
Você é um agente assistente em um programa de detecção de conteúdos falsos (fake news).
|
| 20 |
-
Intereja com o usuário sugerindo as ferramentas que você tem disponível. Se o usuário perguntar do que se trata,
|
| 21 |
-
responda 'Sou uma IA disponível para te ajudar a detectar se um conteúdo que circula na internet é verdadeiro ou não,
|
| 22 |
-
sou capaz de pesquisar na web se o conteúdo é verídico ou não.' Use as ferramenta sempre que possível.
|
| 23 |
-
A data de hoje é .
|
| 24 |
-
Anexe sempre a fonte da pesquisa na resposta (G1, BBC, New York Times, etc), se houver fonte na pesquisa utilizada.
|
| 25 |
-
Não invente fontes, não invente dados. Se não houver nada relacionado à pergunta do usuário nas pesquisas, diga que nada foi encontrado.
|
| 26 |
-
Responda sempre em português do Brasil.
|
| 27 |
-
Limite-se a 600 caracteres.
|
| 28 |
-
"""
|
| 29 |
-
|
| 30 |
-
api_key = os.getenv("GROQ_API_KEY")
|
| 31 |
-
model = GroqModel(
|
| 32 |
-
model_name="openai/gpt-oss-120b"
|
| 33 |
-
)
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
def 'start_convo(user_input, messages_history):
|
| 37 |
-
search_agent = Agent(
|
| 38 |
-
model,
|
| 39 |
-
system_prompt=DEFAULT_SYSTEM_PROMPT
|
| 40 |
-
)
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
return search_agent
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
app.py
CHANGED
|
@@ -1,7 +1,6 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
from ocr_script import ocr_tesseract_only
|
| 3 |
import uuid
|
| 4 |
-
from agent import start_convo
|
| 5 |
|
| 6 |
import os
|
| 7 |
from dotenv import load_dotenv
|
|
@@ -21,9 +20,6 @@ def respond(message, history, user_id, ocr_text):
|
|
| 21 |
system_prompt_text = ocr_text or "Nenhum texto OCR disponível."
|
| 22 |
search_agent = Agent(model, system_prompt=system_prompt_text)
|
| 23 |
|
| 24 |
-
# Se usar seu start_convo, injete o mesmo prompt no agente interno, ou remova se for redundante
|
| 25 |
-
# agent_config = start_convo(user_input=str(message), messages_history=history, system_prompt=system_prompt_text)
|
| 26 |
-
|
| 27 |
result = search_agent.run_sync(str(message))
|
| 28 |
return result.output
|
| 29 |
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
from ocr_script import ocr_tesseract_only
|
| 3 |
import uuid
|
|
|
|
| 4 |
|
| 5 |
import os
|
| 6 |
from dotenv import load_dotenv
|
|
|
|
| 20 |
system_prompt_text = ocr_text or "Nenhum texto OCR disponível."
|
| 21 |
search_agent = Agent(model, system_prompt=system_prompt_text)
|
| 22 |
|
|
|
|
|
|
|
|
|
|
| 23 |
result = search_agent.run_sync(str(message))
|
| 24 |
return result.output
|
| 25 |
|