Spaces:
Running
Running
chore: move llm init to generate prompt step
Browse files- src/agent/image_agent.py +2 -3
- src/agent/music_agent.py +2 -3
src/agent/image_agent.py
CHANGED
|
@@ -66,9 +66,6 @@ class ChangeScene(BaseModel):
|
|
| 66 |
)
|
| 67 |
scene_description: Optional[str] = None
|
| 68 |
|
| 69 |
-
|
| 70 |
-
image_prompt_generator_llm = create_light_llm(0.1).with_structured_output(ChangeScene)
|
| 71 |
-
|
| 72 |
async def generate_image_prompt(user_hash: str, scene_description: str, last_choice = "No choice yet") -> ChangeScene:
|
| 73 |
"""
|
| 74 |
Generates a detailed image prompt string based on a scene description.
|
|
@@ -86,6 +83,8 @@ async def generate_image_prompt(user_hash: str, scene_description: str, last_cho
|
|
| 86 |
last_choice=last_choice,
|
| 87 |
scene_description=scene_description
|
| 88 |
)
|
|
|
|
|
|
|
| 89 |
|
| 90 |
response = await with_retries(lambda: image_prompt_generator_llm.ainvoke(
|
| 91 |
[
|
|
|
|
| 66 |
)
|
| 67 |
scene_description: Optional[str] = None
|
| 68 |
|
|
|
|
|
|
|
|
|
|
| 69 |
async def generate_image_prompt(user_hash: str, scene_description: str, last_choice = "No choice yet") -> ChangeScene:
|
| 70 |
"""
|
| 71 |
Generates a detailed image prompt string based on a scene description.
|
|
|
|
| 83 |
last_choice=last_choice,
|
| 84 |
scene_description=scene_description
|
| 85 |
)
|
| 86 |
+
|
| 87 |
+
image_prompt_generator_llm = create_light_llm(0.1).with_structured_output(ChangeScene)
|
| 88 |
|
| 89 |
response = await with_retries(lambda: image_prompt_generator_llm.ainvoke(
|
| 90 |
[
|
src/agent/music_agent.py
CHANGED
|
@@ -38,9 +38,6 @@ class MusicPrompt(BaseModel):
|
|
| 38 |
prompt: str
|
| 39 |
|
| 40 |
|
| 41 |
-
llm = create_light_llm(0.1).with_structured_output(MusicPrompt)
|
| 42 |
-
|
| 43 |
-
|
| 44 |
async def generate_music_prompt(user_hash: str, scene_description: str, last_choice = "No choice yet") -> str:
|
| 45 |
logger.info(f"Generating music prompt for the current scene: {scene_description}")
|
| 46 |
|
|
@@ -55,6 +52,8 @@ async def generate_music_prompt(user_hash: str, scene_description: str, last_cho
|
|
| 55 |
scene_description=scene_description
|
| 56 |
)
|
| 57 |
|
|
|
|
|
|
|
| 58 |
response = await with_retries(lambda: llm.ainvoke(
|
| 59 |
[SystemMessage(content=system_prompt), HumanMessage(content=scene)]
|
| 60 |
))
|
|
|
|
| 38 |
prompt: str
|
| 39 |
|
| 40 |
|
|
|
|
|
|
|
|
|
|
| 41 |
async def generate_music_prompt(user_hash: str, scene_description: str, last_choice = "No choice yet") -> str:
|
| 42 |
logger.info(f"Generating music prompt for the current scene: {scene_description}")
|
| 43 |
|
|
|
|
| 52 |
scene_description=scene_description
|
| 53 |
)
|
| 54 |
|
| 55 |
+
llm = create_light_llm(0.1).with_structured_output(MusicPrompt)
|
| 56 |
+
|
| 57 |
response = await with_retries(lambda: llm.ainvoke(
|
| 58 |
[SystemMessage(content=system_prompt), HumanMessage(content=scene)]
|
| 59 |
))
|