Skip to content

Commit

Permalink
reformat
Browse files Browse the repository at this point in the history
  • Loading branch information
tylerjthomas9 committed Feb 24, 2024
1 parent 88b9b1b commit 145fa9c
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions test/runtests.jl
Original file line number Diff line number Diff line change
Expand Up @@ -20,10 +20,10 @@ const secret_key = ENV["GOOGLE_API_KEY"]
@test typeof(response) == GoogleGenAI.GoogleTextResponse

# Multi-turn conversation
conversation = [
Dict(:role => "user", :parts => [Dict(:text => "Hello")]),
]
response = generate_content(secret_key, "gemini-pro", conversation; max_output_tokens=50)
conversation = [Dict(:role => "user", :parts => [Dict(:text => "Hello")])]
response = generate_content(
secret_key, "gemini-pro", conversation; max_output_tokens=50
)

n_tokens = count_tokens(secret_key, "gemini-pro", "Hello")
@test n_tokens == 1
Expand Down

0 comments on commit 145fa9c

Please sign in to comment.