Use Anthropic messages API (#304)

This commit is contained in:
Jani Monoses
2024-03-22 22:23:31 +02:00
committed by GitHub
parent 08df63a6f8
commit e57f079275
6 changed files with 25 additions and 19 deletions

View File

@@ -52,7 +52,7 @@ def batch():
if __name__ == "__main__":
sgl.set_default_backend(sgl.Anthropic("claude-2"))
sgl.set_default_backend(sgl.Anthropic("claude-3-haiku-20240307"))
# Run a single request
print("\n========== single ==========\n")

View File

@@ -19,7 +19,7 @@ def few_shot_qa(s, question):
\n\nAssistant: Rome
""")
s += "\n\nHuman: " + question + "\n"
s += "\n\nAssistant:" + sgl.gen("answer", stop="\n", temperature=0)
s += "\n\nAssistant:" + sgl.gen("answer", stop="\\n", temperature=0)
def single():
@@ -52,7 +52,7 @@ def batch():
if __name__ == "__main__":
sgl.set_default_backend(sgl.Anthropic("claude-2"))
sgl.set_default_backend(sgl.Anthropic("claude-3-haiku-20240307"))
# Run a single request
print("\n========== single ==========\n")