Spaces:
Running
Running
Diego Carpintero
commited on
Commit
·
202ed7a
1
Parent(s):
f157f1c
add termination condition for NO_TEXT_FOUND
Browse files- config/agents.yaml +1 -0
- examples/06.landscape.ham.png +0 -0
- examples/log.csv +1 -0
- minerva.py +6 -1
config/agents.yaml
CHANGED
|
@@ -5,6 +5,7 @@ ocr_agent:
|
|
| 5 |
2. Clean and format the extracted text
|
| 6 |
3. Do not perform any analysis on the extracted text
|
| 7 |
4. Reply with the extracted text
|
|
|
|
| 8 |
|
| 9 |
link_checker_agent:
|
| 10 |
assignment: >
|
|
|
|
| 5 |
2. Clean and format the extracted text
|
| 6 |
3. Do not perform any analysis on the extracted text
|
| 7 |
4. Reply with the extracted text
|
| 8 |
+
5. If there is no text in the image, reply with "NO_TEXT_FOUND"
|
| 9 |
|
| 10 |
link_checker_agent:
|
| 11 |
assignment: >
|
examples/06.landscape.ham.png
ADDED
|
examples/log.csv
CHANGED
|
@@ -4,3 +4,4 @@ input_image
|
|
| 4 |
03.en.billing.sms.png
|
| 5 |
04.en.multifactor.email.png
|
| 6 |
05.en.twitter.png
|
|
|
|
|
|
| 4 |
03.en.billing.sms.png
|
| 5 |
04.en.multifactor.email.png
|
| 6 |
05.en.twitter.png
|
| 7 |
+
06.landscape.ham.png
|
minerva.py
CHANGED
|
@@ -2,11 +2,13 @@ from typing import List, AsyncIterator
|
|
| 2 |
import os
|
| 3 |
|
| 4 |
from autogen_agentchat.agents import AssistantAgent
|
|
|
|
| 5 |
from autogen_agentchat.messages import MultiModalMessage
|
| 6 |
from autogen_agentchat.teams import RoundRobinGroupChat
|
| 7 |
from autogen_core import Image as AGImage
|
| 8 |
from autogen_core.tools import FunctionTool
|
| 9 |
from autogen_ext.models.openai import OpenAIChatCompletionClient
|
|
|
|
| 10 |
from dotenv import load_dotenv, find_dotenv
|
| 11 |
from PIL import Image
|
| 12 |
import yaml
|
|
@@ -107,9 +109,12 @@ class Minerva:
|
|
| 107 |
|
| 108 |
def create_team(self) -> RoundRobinGroupChat:
|
| 109 |
"""Create a team of agents that work together in Round Robin fashion"""
|
|
|
|
|
|
|
| 110 |
return RoundRobinGroupChat(
|
| 111 |
self.agents,
|
| 112 |
-
max_turns=6
|
|
|
|
| 113 |
)
|
| 114 |
|
| 115 |
async def reset(self):
|
|
|
|
| 2 |
import os
|
| 3 |
|
| 4 |
from autogen_agentchat.agents import AssistantAgent
|
| 5 |
+
from autogen_agentchat.conditions import TextMentionTermination
|
| 6 |
from autogen_agentchat.messages import MultiModalMessage
|
| 7 |
from autogen_agentchat.teams import RoundRobinGroupChat
|
| 8 |
from autogen_core import Image as AGImage
|
| 9 |
from autogen_core.tools import FunctionTool
|
| 10 |
from autogen_ext.models.openai import OpenAIChatCompletionClient
|
| 11 |
+
|
| 12 |
from dotenv import load_dotenv, find_dotenv
|
| 13 |
from PIL import Image
|
| 14 |
import yaml
|
|
|
|
| 109 |
|
| 110 |
def create_team(self) -> RoundRobinGroupChat:
|
| 111 |
"""Create a team of agents that work together in Round Robin fashion"""
|
| 112 |
+
termination = TextMentionTermination("NO_TEXT_FOUND")
|
| 113 |
+
|
| 114 |
return RoundRobinGroupChat(
|
| 115 |
self.agents,
|
| 116 |
+
max_turns=6,
|
| 117 |
+
termination_condition=termination
|
| 118 |
)
|
| 119 |
|
| 120 |
async def reset(self):
|