Add missing parameter in example

neo4j · Dec 20, 2024 · a15a514 · a15a514
1 parent 81f7ff4
commit a15a514
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/examples/customize/llms/custom_llm.py b/examples/customize/llms/custom_llm.py
@@ -7,7 +7,9 @@
 
 
 class CustomLLM(LLMInterface):
-    def __init__(self, model_name: str, **kwargs: Any):
+    def __init__(
+        self, model_name: str, system_instruction: Optional[str] = None, **kwargs: Any
+    ):
         super().__init__(model_name, **kwargs)
 
     def invoke(