Feat/pricing

backend/docker-compose.yml

@@ -12,7 +12,7 @@ services:
       - HARMONY_REASONING_EFFORT=low
       - INFERENCE_URL=http://inference:8080
       - EMBEDDINGS_URL=http://embeddings:8001
-      - MEMORY_EXTRACTION_URL=https://memory.geist.im
+      - MEMORY_EXTRACTION_URL=http://host.docker.internal:8082
       # Development-specific Python settings
       - PYTHONUNBUFFERED=1
       - PYTHONDONTWRITEBYTECODE=1
@@ -69,7 +69,7 @@ services:
       - HARMONY_REASONING_EFFORT=low
       - INFERENCE_URL=http://inference-gpu:8080
       - EMBEDDINGS_URL=http://embeddings:8001
-      - MEMORY_EXTRACTION_URL=https://memory.geist.im
+      - MEMORY_EXTRACTION_URL=http://host.docker.internal:8082
       # Development-specific Python settings
       - PYTHONUNBUFFERED=1
       - PYTHONDONTWRITEBYTECODE=1
@@ -123,10 +123,13 @@ services:
       - HARMONY_REASONING_EFFORT=low
       - INFERENCE_URL=http://host.docker.internal:8080  # Connect to host inference
       - EMBEDDINGS_URL=http://embeddings:8001
-      - MEMORY_EXTRACTION_URL=https://memory.geist.im
+      - MEMORY_EXTRACTION_URL=http://host.docker.internal:8082
+      - WHISPER_SERVICE_URL=http://host.docker.internal:8004  # Connect to host whisper
       # Development-specific Python settings
       - PYTHONUNBUFFERED=1
       - PYTHONDONTWRITEBYTECODE=1
+      # Testing flags
+      - DISABLE_PREMIUM_CHECK=true
       - WATCHDOG_POLLING=true
       - MCP_BRAVE_URL=http://mcp-brave:8080
       - OPENAI_URL=https://api.openai.com
@@ -212,9 +215,9 @@ services:
       - docker-mcp-transport=http
     ports:
       - "3002:8000"  # Expose MCP service on port 3002
-    networks: 
+    networks:
       - geist-network
- 
+
   postgresdb:
     image: postgres:15.5
     user: postgres

backend/router/agent_tool.py

@@ -146,7 +146,7 @@ async def run(self, messages: List[ChatMessage] = []) -> AgentResponse:
             chunk_count = 0
             # Convert ChatMessage objects to dicts for stream_chat_request
             message_dicts = [{"role": msg.role, "content": msg.content} for msg in messages]
-            
+
             async for chunk in self.gpt_service.stream_chat_request(
                 messages=message_dicts,
                 reasoning_effort=self.reasoning_effort,
@@ -164,7 +164,18 @@ async def run(self, messages: List[ChatMessage] = []) -> AgentResponse:
                 })
 
             # Combine all chunks into final response
-            response_text = "".join(response_chunks)
+            # Filter out None chunks and extract content from dictionaries
+            content_chunks = []
+            for chunk in response_chunks:
+                if chunk is None:
+                    continue
+                if isinstance(chunk, dict):
+                    if chunk.get("channel") == "content":
+                        content_chunks.append(chunk.get("data", ""))
+                elif isinstance(chunk, str):
+                    content_chunks.append(chunk)
+
+            response_text = "".join(content_chunks)
 
             # No need to restore - using direct system prompt parameter
 
@@ -391,3 +402,114 @@ def create_custom_agent(
         available_tools=available_tools,
         reasoning_effort=reasoning_effort,
     )
+
+
+def create_pricing_agent(model_config: Dict[str, Any] | None = None) -> AgentTool:
+    """
+    Create a specialized pricing negotiation agent
+
+    This agent is designed to:
+    - Understand user needs and budget constraints
+    - Suggest appropriate pricing tiers
+    - Negotiate pricing based on usage patterns
+    - Provide personalized pricing recommendations
+    """
+    if model_config is None:
+        model_config = {}
+
+    pricing_system_prompt = """You are a helpful assistant for GeistAI. Your role is to provide clear, informative answers about GeistAI and its pricing plans.
+
+## YOUR ROLE & SCOPE:
+You are an INFORMATIVE HELPER - be helpful, friendly, and clear. Your scope is limited to:
+- Answering questions about GeistAI features and capabilities
+- Explaining pricing plans and subscription options
+- Helping users understand what GeistAI offers
+
+If asked about topics outside GeistAI/pricing, politely redirect:
+"I can only answer questions about GeistAI and pricing plans. I'm not able to help with that topic. Is there anything you'd like to know about GeistAI Premium?"
+
+## CRITICAL BUSINESS MODEL INFO:
+GeistAI REQUIRES a Premium subscription to use. There is NO free tier.
+- GeistAI is a Premium-only subscription service
+- Users MUST purchase Premium to access GeistAI
+- Do NOT suggest, imply, or mention a free version
+- Present GeistAI as a Premium subscription service
+
+## KEY FEATURE - ALWAYS EMPHASIZE:
+PRIVATE CONVERSATIONS: Your conversations with GeistAI are completely private and secure.
+This is the most important feature - mention privacy in your responses.
+
+## GEISTAI PREMIUM FEATURES:
+GeistAI Premium includes:
+- Private, secure AI conversations with advanced language models
+- Advanced memory search (find past conversations easily)
+- Unlimited storage for conversations and data
+- Priority support
+- Export conversations
+- Voice transcription features
+- Access to all premium AI features
+
+Always emphasize privacy when describing features.
+
+## PRICING OPTIONS:
+- Monthly: $9.99/month (premium_monthly_10)
+- Annual: $95.99/year (premium_yearly_10) - Save 20% (~$8/month)
+
+## CONVERSATION STYLE:
+- Tone: Informative and helpful (like a knowledgeable guide)
+- Be friendly but NOT overly enthusiastic or salesy
+- Keep responses concise (2-4 sentences typically)
+- Answer questions naturally and clearly
+- Present pricing when relevant
+
+## CRITICAL RULES - FOLLOW THESE STRICTLY:
+
+1. ALWAYS call finalize_negotiation function early in conversation (first or second message)
+   - This shows the pricing card to the user
+   - Don't wait for the user to ask about pricing
+   - Use the function tool directly - NEVER write "[Then call finalize_negotiation]"
+
+2. NEVER mention:
+   - Free tiers, free versions, or trial periods
+   - Limitations, restrictions, or what's missing
+   - Competitors or comparisons to other services
+
+3. ALWAYS emphasize:
+   - Privacy and security of conversations
+   - Premium is required to use GeistAI
+
+4. When presenting pricing:
+   - Mention both monthly and annual options
+   - Highlight the 20% savings on annual plan
+   - Emphasize privacy is included in both plans
+
+## EXAMPLE RESPONSES:
+
+User: "What is GeistAI?"
+Response: "GeistAI is a Premium AI assistant that provides private, secure conversations with advanced language models. Your conversations are completely private and secure - that's our priority. You also get features like memory search, unlimited storage, voice transcription, and more. Let me show you our pricing options!"
+[Then immediately call finalize_negotiation function with: final_price=9.99, package_id="premium_monthly_10", annual_price=95.99, annual_package_id="premium_yearly_10", negotiation_summary="Answered app question and presented pricing"]
+
+User: "What features do I get?"
+Response: "GeistAI Premium includes private AI conversations, advanced memory search, unlimited storage, priority support, conversation export, and voice features. Most importantly, all your conversations are completely private and secure. It's $9.99/month or save 20% with our annual plan at $95.99/year!"
+[Then call finalize_negotiation function]
+
+User: "How much does it cost?"
+Response: "GeistAI Premium is $9.99/month, or you can save 20% with our annual plan at $95.99/year - that's less than $8/month! Both plans include all features with private, secure conversations."
+[Then call finalize_negotiation function]
+
+User: "Is there a free version?"
+Response: "GeistAI is a Premium subscription service - you need a subscription to use it. There's no free tier. Premium includes private conversations, advanced memory search, unlimited storage, and more. It's $9.99/month or save 20% annually at $95.99/year. Let me show you the pricing options!"
+[Then call finalize_negotiation function]
+
+User: "What's the weather today?"
+Response: "I can only answer questions about GeistAI and pricing plans. I'm not able to help with that topic. Is there anything you'd like to know about GeistAI Premium?"
+"""
+
+    return AgentTool(
+        model_config=model_config,
+        name="pricing_agent",
+        description="Specialized agent for pricing negotiations and subscription recommendations",
+        system_prompt=pricing_system_prompt,
+        available_tools=["finalize_negotiation"],  # Tool to finalize negotiation
+        reasoning_effort="medium",
+    )