klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-02 21:19:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update typing and object assignment for new text to image method return

Browse Source
This commit is contained in:
sabaimran
2023-12-17 21:28:33 +05:30
parent 0288804f2e
commit 03cb86ee46
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/khoj/routers/api.py
View File

@@ -732,7 +732,7 @@ async def chat(
await sync_to_async(save_to_conversation_log)(
q, image, user, meta_log, intent_type="text-to-image", inferred_queries=[improved_image_prompt]
)
content_obj = {"image": image, "intentType": "text-to-image", "inferredQueries": [improved_image_prompt]}
content_obj = {"image": image, "intentType": "text-to-image", "inferredQueries": [improved_image_prompt]} # type: ignore[assignment]
return Response(content=json.dumps(content_obj), media_type="application/json", status_code=status_code)
# Get the (streamed) chat response from the LLM of choice.

2
src/khoj/routers/helpers.py
View File

@@ -266,7 +266,7 @@ def generate_chat_response(
return chat_response, metadata
async def text_to_image(message: str) -> Tuple[Optional[str], int]:
async def text_to_image(message: str) -> Tuple[Optional[str], int, Optional[str]]:
status_code = 200
image = None