Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
  • Loading branch information
sarahwooders committed Jun 5, 2024
1 parent a8c7531 commit b385fc4
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions tests/test_endpoints.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@
embedding_config_dir = "configs/embedding_model_configs"
llm_config_dir = "configs/llm_model_configs"


def run_llm_endpoint(filename):
config_data = json.load(open(filename, "r"))
print(config_data)
Expand Down Expand Up @@ -83,12 +84,12 @@ def test_embedding_endpoint_local():
filename = os.path.join(embedding_config_dir, "local.json")
run_embedding_endpoint(filename)


def test_llm_endpoint_ollama():
filename = os.path.join(llm_config_dir, "ollama.json")
run_llm_endpoint(filename)


def test_embedding_endpoint_ollama():
filename = os.path.join(embedding_config_dir, "ollama.json")
run_embedding_endpoint(filename)


0 comments on commit b385fc4

Please sign in to comment.