Adding Autocomplete to OSS (#1198)

* Cleaned up model_config

* Fix pydantic issues

* 99% done with autocomplete

* fixed test issues

* Fix type checking issues
This commit is contained in:
Brandon Hancock (bhancock_ai)
2024-08-16 15:04:21 -04:00
committed by GitHub
parent 3451b6fc7a
commit bf7372fefa
14 changed files with 110 additions and 122 deletions

View File

@@ -1,8 +1,8 @@
"""Test Agent creation and execution basic functionality."""
import os
import hashlib
import json
import os
from unittest.mock import MagicMock, patch
import pytest
@@ -703,7 +703,7 @@ def test_task_definition_based_on_dict():
"expected_output": "The score of the title.",
}
task = Task(config=config)
task = Task(**config)
assert task.description == config["description"]
assert task.expected_output == config["expected_output"]
@@ -716,7 +716,7 @@ def test_conditional_task_definition_based_on_dict():
"expected_output": "The score of the title.",
}
task = ConditionalTask(config=config, condition=lambda x: True)
task = ConditionalTask(**config, condition=lambda x: True)
assert task.description == config["description"]
assert task.expected_output == config["expected_output"]