Minor Updates

This commit is contained in:
2024-10-31 16:46:18 +01:00
parent 1fe15cf83a
commit e9893f125c
6 changed files with 18 additions and 9 deletions

View File

@ -13,8 +13,7 @@ import re
from typing import Callable
from ums.agent import ExtractAudioAgent, ExtractImageAgent, ExtractTextAgent
from ums.utils.types import RiddleData, AgentMessage
from ums.utils.schema import ExtractedData, ExtractedContent, ExtractedPositions
from ums.utils import RiddleData, AgentMessage, ExtractedData, ExtractedContent, ExtractedPositions
class SimpleExtractAudioAgent(ExtractAudioAgent):
# here we do not have an implementation for extracting audio,

View File

@ -14,7 +14,7 @@ import re, random
from typing import Callable
from ums.agent import SolveAgent
from ums.utils.types import Riddle, RiddleData, RiddleSolution, RiddleDataType, RiddleStatus, AgentMessage
from ums.utils import Riddle, RiddleData, RiddleSolution, RiddleDataType, AgentMessage
class SimpleSolveAgent(SolveAgent):

View File

@ -12,7 +12,7 @@ import re
from typing import Callable
from ums.agent import GatekeeperAgent
from ums.utils.types import Riddle, RiddleSolution, AgentMessage
from ums.utils import Riddle, RiddleSolution, AgentMessage
class SimpleSolveAgent(GatekeeperAgent):