Skip to content

Commit

Permalink
fix cli tests
Browse files Browse the repository at this point in the history
  • Loading branch information
ThejasNU committed Jan 28, 2025
1 parent 7ff3876 commit 38d4baa
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions libs/agentc_testing/agentc_testing/repo.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,6 +73,9 @@ def initialize_repo(
repo.index.commit("Initial commit")
output = list()

# Initialize the local catalog.
click_runner.invoke(click_command, ["init", "local", "catalog"])

# If we are not using the index command, we can return early...
if repo_kind == ExampleRepoKind.EMPTY or repo_kind == ExampleRepoKind.NON_INDEXED_ALL_TRAVEL:
return output
Expand All @@ -90,6 +93,9 @@ def initialize_repo(
]:
return output

# Initialize the DB catalog.
click_runner.invoke(click_command, ["init", "db", "catalog", "--bucket", "travel-sample"])

# Call our publish command. Note that this assumes a container / CB instance is active!
os.environ["AGENT_CATALOG_MAX_SOURCE_PARTITION"] = "1"
os.environ["AGENT_CATALOG_INDEX_PARTITION"] = "1"
Expand Down

0 comments on commit 38d4baa

Please sign in to comment.