import neo4j
from langchain_openai.chat_models import ChatOpenAI
from langchain_openai.embeddings import OpenAIEmbeddings
from neo4j_graphrag.generation import GraphRAG
from neo4j_graphrag.retrievers import VectorCypherRetriever
import os



# Define database credentials
URI  = "neo4j://x.x.x.x:19009"
AUTH = ("neo4j", "xxxx")
DATABASE = "neo4j"
INDEX = "xxx_embeddings"


driver = neo4j.GraphDatabase.driver(
    URI,
    auth=AUTH,
)

embedder = OpenAIEmbeddings(model="text-embedding-ada-002")

retriever = VectorCypherRetriever(
    driver,
    index_name=INDEX,
    retrieval_query="WITH node, score RETURN node.title as title, node.plot as plot",
    embedder=embedder,  # type: ignore[arg-type, unused-ignore]
    neo4j_database=DATABASE,
)

llm = ChatOpenAI(model="gpt-4o", temperature=0)

rag = GraphRAG(
    retriever=retriever,
    llm=llm,  # type: ignore[arg-type, unused-ignore]
)

result = rag.search(
    "have movie?",
    return_context=False,
)
print(result.answer)

driver.close()

Traceback (most recent call last):
  File "/Users/heishen/Desktop/work_Mac_mini/Developer/GraphRAG_/langchain 测试.py", line 49, in <module>
    result = rag.search(
             ^^^^^^^^^^^
  File "/opt/anaconda3/envs/test/lib/python3.11/site-packages/neo4j_graphrag/generation/graphrag.py", line 150, in search
    llm_response = self.llm.invoke(
                   ^^^^^^^^^^^^^^^^
  File "/opt/anaconda3/envs/test/lib/python3.11/site-packages/langchain_core/language_models/chat_models.py", line 378, in invoke
    self.generate_prompt(
  File "/opt/anaconda3/envs/test/lib/python3.11/site-packages/langchain_core/language_models/chat_models.py", line 963, in generate_prompt
    return self.generate(prompt_messages, stop=stop, callbacks=callbacks, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/anaconda3/envs/test/lib/python3.11/site-packages/langchain_core/language_models/chat_models.py", line 782, in generate
    self._generate_with_cache(
  File "/opt/anaconda3/envs/test/lib/python3.11/site-packages/langchain_core/language_models/chat_models.py", line 1028, in _generate_with_cache
    result = self._generate(
             ^^^^^^^^^^^^^^^
  File "/opt/anaconda3/envs/test/lib/python3.11/site-packages/langchain_openai/chat_models/base.py", line 1131, in _generate
    response = self.client.create(**payload)
               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/anaconda3/envs/test/lib/python3.11/site-packages/openai/_utils/_utils.py", line 287, in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
TypeError: Completions.create() got an unexpected keyword argument 'system_instruction'
(test) heishen@macbook-pro GraphRAG_ %