26
examples/quick_start/gemini_example_complete.py
Normal file
26
examples/quick_start/gemini_example_complete.py
Normal file
@@ -0,0 +1,26 @@
|
||||
from sglang import function, gen, set_default_backend, Gemini
|
||||
|
||||
|
||||
@function
|
||||
def few_shot_qa(s, question):
|
||||
s += (
|
||||
"""The following are questions with answers.
|
||||
Q: What is the capital of France?
|
||||
A: Paris
|
||||
Q: What is the capital of Germany?
|
||||
A: Berlin
|
||||
Q: What is the capital of Italy?
|
||||
A: Rome
|
||||
""")
|
||||
s += "Q: " + question + "\n"
|
||||
s += "A:" + gen("answer", stop="\n", temperature=0)
|
||||
|
||||
|
||||
set_default_backend(Gemini("gemini-pro"))
|
||||
|
||||
state = few_shot_qa.run(question="What is the capital of the United States?")
|
||||
answer = state["answer"].strip().lower()
|
||||
|
||||
assert "washington" in answer, f"answer: {state['answer']}"
|
||||
|
||||
print(state.text())
|
||||
Reference in New Issue
Block a user