mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-11 17:18:29 +00:00
e16606672a
git-subtree-dir: packages/tools git-subtree-split: 78317b9c127f18bd040c1d77e3c0840cdc9a5b38
22 lines
506 B
Python
22 lines
506 B
Python
from typing import Callable
|
|
|
|
import pytest
|
|
|
|
|
|
class Helpers:
|
|
@staticmethod
|
|
def get_embedding_function() -> Callable:
|
|
def _func(input):
|
|
assert input == ["What are the requirements for the task?"]
|
|
with open("tests/data/embedding.txt", "r") as file:
|
|
content = file.read()
|
|
numbers = content.split(",")
|
|
return [[float(number) for number in numbers]]
|
|
|
|
return _func
|
|
|
|
|
|
@pytest.fixture
|
|
def helpers():
|
|
return Helpers
|