package seems to work

This commit is contained in:
Moishe Lettvin
2023-12-27 12:17:00 -05:00
parent 2d24309a94
commit 9e9007f7b7
29 changed files with 42 additions and 75 deletions

View File

@@ -1,50 +0,0 @@
import os
import random
import time
"""
from algoliasearch.configs import SearchConfig
from algoliasearch.search_client import SearchClient
"""
class SearchIndexer():
def __init__(self, story_id):
pass
def index_text(self, text):
pass
def index_image(self, text):
pass
"""
class AlgoliaSearchIndexer(SearchIndexer):
def __init__(self, story_id):
self.index = None
self.story_id = story_id
self.search_enabled = os.getenv('ALGOLIA_APP_ID') and os.getenv('ALGOLIA_API_KEY')
if self.search_enabled:
config = SearchConfig(os.getenv('ALGOLIA_APP_ID'), os.getenv('ALGOLIA_API_KEY'))
self.algolia = SearchClient.create_with_config(config)
self.index = self.algolia.init_index('daily-llm-conversations')
def index_text(self, text):
if self.index:
res = self.index.save_object({
"objectID": hex(random.getrandbits(128))[2:],
"storyID": self.story_id,
"type": "text",
"text": text,
"createdAt": int(time.time())
}).wait()
def index_image(self, url):
if self.index:
self.index.save_object({
"objectID": hex(random.getrandbits(128))[2:],
"storyID": self.story_id,
"type": "image",
"image": url,
"createdAt": int(time.time())
}).wait()
"""

View File

View File

@@ -9,12 +9,11 @@ from threading import Event, Semaphore, Thread
from typing import Iterator, Optional, Type, TypedDict
from typing_extensions import Unpack
from services.ai_services import AIServiceConfig
from message_handler.message_handler import MessageHandler
from dailyai.services.ai_services import AIServiceConfig
from dailyai.message_handler.message_handler import MessageHandler
frame_idx = 0
class AsyncProcessorState:
# Setting class variables, other synchronous activities
INIT = 0

View File

@@ -1,5 +1,5 @@
Metadata-Version: 2.1
Name: daily_ai
Name: dailyai
Version: 0.0.1
Summary: Orchestrator for AI bots with Daily
Requires-Dist: daily-python

View File

@@ -1,11 +1,11 @@
README.md
pyproject.toml
src/daily-ai/async_processor/async_processor.py
src/daily-ai/daily_ai.egg-info/PKG-INFO
src/daily-ai/daily_ai.egg-info/SOURCES.txt
src/daily-ai/daily_ai.egg-info/dependency_links.txt
src/daily-ai/daily_ai.egg-info/requires.txt
src/daily-ai/daily_ai.egg-info/top_level.txt
src/daily-ai/dailyai.egg-info/PKG-INFO
src/daily-ai/dailyai.egg-info/SOURCES.txt
src/daily-ai/dailyai.egg-info/dependency_links.txt
src/daily-ai/dailyai.egg-info/requires.txt
src/daily-ai/dailyai.egg-info/top_level.txt
src/daily-ai/message_handler/message_handler.py
src/daily-ai/services/ai_services.py
src/daily-ai/services/azure_ai_services.py
@@ -19,4 +19,4 @@ src/daily-ai/services/open_ai_service.py
src/daily-ai/services/playht_ai_service.py
src/daily-ai/storage/search.py
src/daily-ai/tests/test_asyncprocessor.py
src/daily-ai/tests/test_message_handler.py
src/daily-ai/tests/test_message_handler.py

View File

View File

@@ -5,8 +5,8 @@ from dataclasses import dataclass
from queue import Queue, Empty
from threading import Thread
from storage.search import SearchIndexer
from services.ai_services import AIServiceConfig
from dailyai.storage.search import SearchIndexer
from dailyai.services.ai_services import AIServiceConfig
@dataclass

View File

@@ -6,14 +6,14 @@ import wave
from dataclasses import dataclass
from queue import Queue, Empty
from daily_ai.async_processor import (
from dailyai.async_processor.async_processor import (
AsyncProcessor,
AsyncProcessorState,
ConversationProcessorCollection,
Response,
)
from daily_ai.services.ai_services import AIServiceConfig
from daily_ai.message_handler import MessageHandler
from dailyai.services.ai_services import AIServiceConfig
from dailyai.message_handler.message_handler import MessageHandler
from threading import Thread, Semaphore, Event, Timer

View File

View File

@@ -6,7 +6,7 @@ import requests
from typing import Generator
from daily_ai.services.ai_services import LLMService, TTSService, ImageGenService
from dailyai.services.ai_services import LLMService, TTSService, ImageGenService
from PIL import Image
# See .env.example for Azure configuration needed

View File

@@ -4,7 +4,7 @@ import time
from typing import Generator
from daily_ai.services.ai_services import TTSService
from ..services.ai_services import TTSService
class ElevenLabsTTSService(TTSService):

View File

View File

@@ -0,0 +1,9 @@
class SearchIndexer():
def __init__(self, story_id):
pass
def index_text(self, text):
pass
def index_image(self, text):
pass

View File

View File

@@ -5,13 +5,17 @@ from queue import Queue, Empty
from threading import Thread, Event
from typing import Generator
from services.ai_services import LLMService, TTSService, ImageGenService
from message_handler.message_handler import MessageHandler
from async_processor.async_processor import (
AsyncProcessor,
from dailyai.services.ai_services import (
AIServiceConfig,
ImageGenService,
LLMService,
TTSService
)
from dailyai.message_handler.message_handler import MessageHandler
from dailyai.async_processor.async_processor import (
AsyncProcessor,
AsyncProcessorState,
Response
Response,
)
class MockTTSService(TTSService):

View File

@@ -3,9 +3,14 @@ import unittest
from unittest.mock import MagicMock, call
from message_handler.message_handler import MessageHandler, IndexingMessageHandler
from services.ai_services import AIService, AIServiceConfig, TTSService, LLMService, ImageGenService
from storage.search import SearchIndexer
from dailyai.message_handler.message_handler import MessageHandler, IndexingMessageHandler
from dailyai.services.ai_services import (
AIServiceConfig,
TTSService,
LLMService,
ImageGenService,
)
from ..storage.search import SearchIndexer
class TestMessageHandler(unittest.TestCase):