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,