Skip to content
Open
Changes from 1 commit
Commits
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: 2 additions & 2 deletions examples/interior_design_assistant/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
import uuid
from pathlib import Path
from typing import List

from examples.interior_design_assistant.utils import enforce_response_format
Copy link
Contributor

@yanxi0830 yanxi0830 Nov 22, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

broken import here?

import fire

from examples.interior_design_assistant.utils import (
Expand Down Expand Up @@ -168,7 +168,7 @@ async def suggest_alternatives(

result = turn.output_message.content
print(result)
return [r["description"].strip() for r in json.loads(result.strip())]
return enforce_response_format(result, n)

async def retrieve_images(self, description: str) -> List[ImageMedia]:
"""
Expand Down