Align expected types to actual returned types by AI APIs, operator

src/khoj/processor/conversation/anthropic/anthropic_chat.py

@@ -164,7 +164,7 @@ async def converse_anthropic(
     generated_asset_results: Dict[str, Dict] = {},
     deepthought: Optional[bool] = False,
     tracer: dict = {},
-) -> AsyncGenerator[ResponseWithThought, None]:
+) -> AsyncGenerator[str | ResponseWithThought, None]:
     """
     Converse with user using Anthropic's Claude
     """

src/khoj/processor/conversation/openai/gpt.py

@@ -190,7 +190,7 @@ async def converse_openai(
     program_execution_context: List[str] = None,
     deepthought: Optional[bool] = False,
     tracer: dict = {},
-) -> AsyncGenerator[ResponseWithThought, None]:
+) -> AsyncGenerator[str | ResponseWithThought, None]:
     """
     Converse with user using OpenAI's ChatGPT
     """

src/khoj/processor/operator/grounding_agent_uitars.py

@@ -13,7 +13,7 @@ from io import BytesIO
 from typing import Any, List
 
 import numpy as np
-from openai import AzureOpenAI, OpenAI
+from openai import AsyncAzureOpenAI, AsyncOpenAI
 from openai.types.chat import ChatCompletion
 from PIL import Image
 
@@ -72,7 +72,7 @@ class GroundingAgentUitars:
     def __init__(
         self,
         model_name: str,
-        client: OpenAI | AzureOpenAI,
+        client: AsyncOpenAI | AsyncAzureOpenAI,
         max_iterations=50,
         environment_type: Literal["computer", "web"] = "computer",
         runtime_conf: dict = {

src/khoj/routers/api_model.py

@@ -72,7 +72,7 @@ async def update_chat_model(
     if chat_model is None:
         return Response(status_code=404, content=json.dumps({"status": "error", "message": "Chat model not found"}))
     if not subscribed and chat_model.price_tier != PriceTier.FREE:
-        raise Response(
+        return Response(
             status_code=403,
             content=json.dumps({"status": "error", "message": "Subscribe to switch to this chat model"}),
         )
@@ -108,7 +108,7 @@ async def update_voice_model(
     if voice_model is None:
         return Response(status_code=404, content=json.dumps({"status": "error", "message": "Voice model not found"}))
     if not subscribed and voice_model.price_tier != PriceTier.FREE:
-        raise Response(
+        return Response(
             status_code=403,
             content=json.dumps({"status": "error", "message": "Subscribe to switch to this voice model"}),
         )
@@ -143,7 +143,7 @@ async def update_paint_model(
     if image_model is None:
         return Response(status_code=404, content=json.dumps({"status": "error", "message": "Image model not found"}))
     if not subscribed and image_model.price_tier != PriceTier.FREE:
-        raise Response(
+        return Response(
             status_code=403,
             content=json.dumps({"status": "error", "message": "Subscribe to switch to this image model"}),
         )