Skip to content

Commit

Permalink
Fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
krassowski committed Apr 5, 2024
1 parent 2067552 commit caa6fa8
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,10 @@
dict({
'api_keys': list([
]),
'completions_embeddings_provider_id': None,
'completions_fields': dict({
}),
'completions_model_provider_id': None,
'embeddings_provider_id': None,
'fields': dict({
}),
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,8 @@ def __init__(self):
self.messages = []
self.tasks = []
self.settings["jai_config_manager"] = SimpleNamespace(
lm_provider=MockProvider, lm_provider_params={"model_id": "model"}
completions_lm_provider=MockProvider,
completions_lm_provider_params={"model_id": "model"},
)
self.settings["jai_event_loop"] = SimpleNamespace(
create_task=lambda x: self.tasks.append(x)
Expand Down

0 comments on commit caa6fa8

Please sign in to comment.