Skip to content

Commit 97a6613

Browse files
committed
fix: cls.prompt is a string, use directly
1 parent 9ed8210 commit 97a6613

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

dingo/model/llm/hhh/llm_text_3h.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,7 @@ class LLMText3H(BaseOpenAI):
1313
def build_messages(cls, input_data):
1414
question = input_data.prompt
1515
response = input_data.content
16-
# cls.prompt may be a string or a class with .content attribute
17-
prompt_template = getattr(cls.prompt, 'content', cls.prompt)
18-
prompt_content = prompt_template % (question, response)
16+
prompt_content = cls.prompt % (question, response)
1917

2018
messages = [{"role": "user", "content": prompt_content}]
2119

0 commit comments

Comments
 (0)