| import anthropic |
| |
| prompt_template = "Tell me a short joke about {topic}" |
| anthropic_template = f"Human:\n\n{prompt_template}\n\nAssistant:" |
| anthropic_client = anthropic.Anthropic() |
| |
| |
| def call_anthropic(prompt_value: str) -> str: |
| response = anthropic_client.completions.create( |
| model="claude-2", |
| prompt=prompt_value, |
| max_tokens_to_sample=256, |
| ) |
| return response.completion |
| |
| |
| def invoke_anthropic_chain(topic: str) -> str: |
| prompt_value = anthropic_template.format(topic=topic) |
| return call_anthropic(prompt_value) |
| |
| |
| if __name__ == "__main__": |
| print(invoke_anthropic_chain("ice cream")) |
