Skip to content

Commit

Permalink
Added agent_test_settings to actually get data files
Browse files Browse the repository at this point in the history
  • Loading branch information
jamesbraza committed Sep 13, 2024
1 parent 04c7c4d commit 8736e3d
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions tests/test_agents.py
Original file line number Diff line number Diff line change
Expand Up @@ -110,7 +110,7 @@ async def test_agent_types(


@pytest.mark.asyncio
async def test_successful_memory_agent() -> None:
async def test_successful_memory_agent(agent_test_settings: Settings) -> None:
tic = time.perf_counter()
memory_id = "call_Wtmv95JbNcQ2nRQCZBoOfcJy" # Stub value
memory = Memory(
Expand Down Expand Up @@ -148,7 +148,10 @@ async def test_successful_memory_agent() -> None:
)
await memory_model.add_memory(memory)
serialized_memory_model = memory_model.model_dump(exclude_none=True)
query = QueryRequest(query="How can you use XAI for chemical property prediction?")
query = QueryRequest(
query="How can you use XAI for chemical property prediction?",
settings=agent_test_settings,
)
# NOTE: use Claude 3 for its <thinking> feature, testing regex replacement of it
query.settings.agent.agent_llm = "claude-3-opus-20240229"
query.settings.agent.agent_config = {
Expand Down

0 comments on commit 8736e3d

Please sign in to comment.