13 lines
365 B
Python
13 lines
365 B
Python
|
from neo4j_graphrag.llm import AnthropicLLM, LLMResponse
|
||
|
|
|
||
|
|
# set api key here on in the ANTHROPIC_API_KEY env var
|
||
|
|
api_key = None
|
||
|
|
|
||
|
|
with AnthropicLLM(
|
||
|
|
model_name="claude-3-opus-20240229",
|
||
|
|
model_params={"max_tokens": 1000}, # max_tokens must be specified
|
||
|
|
api_key=api_key,
|
||
|
|
) as llm:
|
||
|
|
res: LLMResponse = llm.invoke("say something")
|
||
|
|
print(res.content)
|