Skip to content

Commit e22bc7c

Browse files
Adequa uso do serviço Llama em tasks
1 parent 1fbd7d1 commit e22bc7c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

markup_doc/tasks.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -151,10 +151,10 @@ def get_labels(title, user_id):
151151
continue
152152

153153
if item.get('type') == 'first_block':
154-
first_block = GenericLlama(type='prompt', temperature=0.1)
154+
first_block = LlamaService(type='prompt', temperature=0.1)
155155

156156
if getLLM() == 'GEMINI':
157-
output = first_block.run(functionsLlama.getFirstMetadata(clean_labels(item.get('text'))))
157+
output = first_block.run(LlamaInputSettings.get_first_metadata(clean_labels(item.get('text'))))
158158
match = re.search(r'\{.*\}', output, re.DOTALL)
159159
if match:
160160
output = match.group(0)

0 commit comments

Comments
 (0)