mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-22 14:28:55 +00:00
First take on a rag tool
This commit is contained in:
21
tests/conftest.py
Normal file
21
tests/conftest.py
Normal file
@@ -0,0 +1,21 @@
|
||||
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
|
||||
Reference in New Issue
Block a user