mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-05-19 00:00:37 +02:00
25 lines
708 B
Python
25 lines
708 B
Python
from langchain_core.messages import HumanMessage
|
|
|
|
from onyx.llm.utils import build_content_with_imgs
|
|
|
|
|
|
IMG_GENERATION_SUMMARY_PROMPT = """
|
|
You have just created the attached images in response to the following query: "{query}".
|
|
|
|
Can you please summarize them in a sentence or two? Do NOT include image urls or bulleted lists.
|
|
"""
|
|
|
|
|
|
def build_image_generation_user_prompt(
|
|
query: str,
|
|
img_urls: list[str] | None = None,
|
|
b64_imgs: list[str] | None = None,
|
|
) -> HumanMessage:
|
|
return HumanMessage(
|
|
content=build_content_with_imgs(
|
|
message=IMG_GENERATION_SUMMARY_PROMPT.format(query=query).strip(),
|
|
b64_imgs=b64_imgs,
|
|
img_urls=img_urls,
|
|
)
|
|
)
|