diff --git a/evals/completion_fns/uni_finder.py b/evals/completion_fns/uni_finder.py index 5a6fa04c9e..363eb48fc2 100644 --- a/evals/completion_fns/uni_finder.py +++ b/evals/completion_fns/uni_finder.py @@ -66,7 +66,7 @@ def __call__(self, prompt: Union[str, list[dict]], **kwargs: Any) -> UniFinderCo url = f"{self.api_base}/api/external/upload_pdf" files = {'file': open(kwargs["file_name"], 'rb')} data = { - 'pdf_parse_mode': self.pdf_parse_mode, + # 'pdf_parse_mode': self.pdf_parse_mode, 'api_key': self.api_key } response = requests.post(url, data=data, files=files) diff --git a/evals/registry/completion_fns/uni_finder.yaml b/evals/registry/completion_fns/uni_finder.yaml index ae2c7b778e..e97e4c0310 100644 --- a/evals/registry/completion_fns/uni_finder.yaml +++ b/evals/registry/completion_fns/uni_finder.yaml @@ -1,23 +1,4 @@ -uni_finder/fast/gpt-3.5-turbo: +uni_finder-v1.26: class: evals.completion_fns.uni_finder:UniFinderCompletionFn args: - pdf_parse_mode: fast - model: gpt35 - -uni_finder/precise/gpt-3.5-turbo: - class: evals.completion_fns.uni_finder:UniFinderCompletionFn - args: - pdf_parse_mode: precise - model: gpt35 - -uni_finder/fast/gpt-4-all: - class: evals.completion_fns.uni_finder:UniFinderCompletionFn - args: - pdf_parse_mode: fast - model: gpt4 - -uni_finder/precise/gpt-4-all: - class: evals.completion_fns.uni_finder:UniFinderCompletionFn - args: - pdf_parse_mode: precise - model: gpt4 \ No newline at end of file + pdf_parse_mode: v1.26 \ No newline at end of file