Skip to content

feat: backend for floating bar for macos #3407

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
arsenkylyshbek wants to merge 1 commit into from
+854 −84
Closed

feat: backend for floating bar for macos #3407

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions backend/.env.template
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -55,3 +55,7 @@ APPLE_TEAM_ID=
APPLE_KEY_ID=
APPLE_PRIVATE_KEY=
API_BASE_URL=

PERPLEXITY_API_KEY=

BUCKET_CHAT_FILES=
29 changes: 22 additions & 7 deletions backend/routers/chat.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@
from multipart.multipart import shutil

import database.chat as chat_db
import database.conversations as conversations_db
from database.apps import record_app_usage
from models.app import App, UsageHistoryType
from models.chat import (
Expand All @@ -34,6 +35,7 @@
from utils.other import endpoints as auth, storage
from utils.other.chat_file import FileChatTool
from utils.retrieval.graph import execute_graph_chat, execute_graph_chat_stream, execute_persona_chat_stream
from utils.retrieval.agentic import execute_agentic_chat, execute_agentic_chat_stream

router = APIRouter()

Expand Down Expand Up @@ -82,10 +84,16 @@ def send_message(
type='text',
app_id=compat_app_id,
)
if data.file_ids is not None and chat_session:
new_file_ids = chat_session.retrieve_new_file(data.file_ids)
chat_session.add_file_ids(data.file_ids)
chat_db.add_files_to_chat_session(uid, chat_session.id, data.file_ids)

# Handle file attachments
if data.file_ids is not None and len(data.file_ids) > 0:
if chat_session:
new_file_ids = chat_session.retrieve_new_file(data.file_ids)
chat_session.add_file_ids(data.file_ids)
chat_db.add_files_to_chat_session(uid, chat_session.id, data.file_ids)
else:
# No chat session, use all file_ids
new_file_ids = data.file_ids

if len(new_file_ids) > 0:
message.files_id = new_file_ids
Expand All @@ -99,12 +107,18 @@ def send_message(

chat_db.add_message(uid, message.dict())

app = get_available_app_by_id(compat_app_id, uid)
app = App(**app) if app else None
# Try to get app, but don't fail if Redis is down (local dev)
try:
app = get_available_app_by_id(compat_app_id, uid)
app = App(**app) if app else None
except Exception as e:
print(f"⚠️ Could not get app from cache (Redis issue): {e}")
print("⚠️ Continuing without app (local dev mode)")
app = None

app_id_from_app = app.id if app else None

messages = list(reversed([Message(**msg) for msg in chat_db.get_messages(uid, limit=10, app_id=compat_app_id)]))
messages = list(reversed([Message(**msg) for msg in chat_db.get_messages(uid, limit=10, app_id=compat_app_id, include_conversations=True)]))

def process_message(response: str, callback_data: dict):
memories = callback_data.get('memories_found', [])
Expand Down Expand Up @@ -149,6 +163,7 @@ def process_message(response: str, callback_data: dict):

async def generate_stream():
callback_data = {}
# Using the new agentic system via graph routing
async for chunk in execute_graph_chat_stream(
uid, messages, app, cited=True, callback_data=callback_data, chat_session=chat_session
):
Expand Down
Loading