mirror of
https://github.com/lfnovo/open-notebook.git
synced 2026-05-02 13:20:30 +00:00
d7b0fff954
Creates the API layer for Open Notebook Creates a services API gateway for the Streamlit front-end Migrates the SurrealDB SDK to the official one Change all database calls to async New podcast framework supporting multiple speaker configurations Implement the surreal-commands library for async processing Improve docker image and docker-compose configurations
56 lines
No EOL
1.4 KiB
Python
56 lines
No EOL
1.4 KiB
Python
"""
|
|
Search service layer using API.
|
|
"""
|
|
|
|
from typing import Dict, List, Any
|
|
|
|
from loguru import logger
|
|
|
|
from api.client import api_client
|
|
|
|
|
|
class SearchService:
|
|
"""Service layer for search operations using API."""
|
|
|
|
def __init__(self):
|
|
logger.info("Using API for search operations")
|
|
|
|
def search(
|
|
self,
|
|
query: str,
|
|
search_type: str = "text",
|
|
limit: int = 100,
|
|
search_sources: bool = True,
|
|
search_notes: bool = True,
|
|
minimum_score: float = 0.2
|
|
) -> List[Dict[str, Any]]:
|
|
"""Search the knowledge base."""
|
|
response = api_client.search(
|
|
query=query,
|
|
search_type=search_type,
|
|
limit=limit,
|
|
search_sources=search_sources,
|
|
search_notes=search_notes,
|
|
minimum_score=minimum_score
|
|
)
|
|
return response.get("results", [])
|
|
|
|
def ask_knowledge_base(
|
|
self,
|
|
question: str,
|
|
strategy_model: str,
|
|
answer_model: str,
|
|
final_answer_model: str
|
|
) -> Dict[str, str]:
|
|
"""Ask the knowledge base a question."""
|
|
response = api_client.ask_simple(
|
|
question=question,
|
|
strategy_model=strategy_model,
|
|
answer_model=answer_model,
|
|
final_answer_model=final_answer_model
|
|
)
|
|
return response
|
|
|
|
|
|
# Global service instance
|
|
search_service = SearchService() |