“ydy0615”
17d211bf93
- Introduced a new web search block plugin to handle web search queries and results. - Updated copilot, doc block, and pro block plugins to include web search context in AI completions. - Implemented utility functions for parsing and building web search markdown. - Enhanced API to support web search requests and responses. - Added configuration for web search URL and timeout settings. - Updated size limit checks to account for web search content.
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
import asyncio
|
|
import logging
|
|
|
|
from job_handlers import (
|
|
asr_handler,
|
|
completion_handler,
|
|
compress_handler,
|
|
convert_handler,
|
|
ocr_handler,
|
|
pro_completion_handler,
|
|
tts_handler,
|
|
web_search_handler,
|
|
)
|
|
from job_system import RedisJobManager, RedisWorker
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format="%(asctime)s %(levelname)s %(name)s - %(message)s",
|
|
)
|
|
logger = logging.getLogger("worker")
|
|
|
|
|
|
async def main() -> None:
|
|
manager = RedisJobManager()
|
|
manager.register_handler("completion", completion_handler)
|
|
manager.register_handler("pro_completion", pro_completion_handler)
|
|
manager.register_handler("web_search", web_search_handler)
|
|
manager.register_handler("compress", compress_handler)
|
|
manager.register_handler("ocr", ocr_handler)
|
|
manager.register_handler("convert", convert_handler)
|
|
manager.register_handler("tts", tts_handler)
|
|
manager.register_handler("asr", asr_handler)
|
|
worker = RedisWorker(manager)
|
|
try:
|
|
await worker.run_forever()
|
|
finally:
|
|
await manager.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|