Skip to content

Commit

Permalink
Merge branch 'support-anyscale-openai-models' of github.com:whitead/p…
Browse files Browse the repository at this point in the history
…aper-qa into support-anyscale-openai-models
  • Loading branch information
Michael Skarlinski committed Jun 14, 2024
2 parents e4ef25b + 3462b5e commit 8ffb10a
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions tests/test_paperqa.py
Original file line number Diff line number Diff line change
Expand Up @@ -580,11 +580,7 @@ async def test_anyscale_chain():
skip_system=True,
)
outputs = []

def accum(x):
outputs.append(x)

completion = await call({"animal": "duck"}, callbacks=[accum]) # type: ignore[call-arg]
completion = await call({"animal": "duck"}, callbacks=[outputs.append]) # type: ignore[call-arg]
assert completion.seconds_to_first_token > 0
assert completion.prompt_count > 0
assert completion.completion_count > 0
Expand Down

0 comments on commit 8ffb10a

Please sign in to comment.