Skip to content

Commit

Permalink
fix sdk tests
Browse files Browse the repository at this point in the history
  • Loading branch information
ThejasNU committed Feb 6, 2025
1 parent ad1e791 commit 3f18f5f
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -51,8 +51,8 @@ def {{ tool.name }}(question: {{ input.type_name }}) -> list[str]:
else:
raise ValueError("Bad input given to tool!")

if
import sentence_transformers

logger.debug("{{ tool.name }} is generating an embedding for: " + formatted_question + ".")
embedding_model = sentence_transformers.SentenceTransformer(
'{{ vector_search.embedding_model.name }}',
Expand Down

0 comments on commit 3f18f5f

Please sign in to comment.