add

agentgraph/extraction/graph_processing/knowledge_graph_processor.py

@@ -122,7 +122,22 @@ class SlidingWindowMonitor:
         
         # Initialize extraction method
         from agentgraph.shared.extraction_factory import create_extraction_method, method_requires_line_numbers, method_requires_content_references
-        self.extraction_method = create_extraction_method(method_name)
+        
+        # Pass model parameter to extraction method based on method type
+        if method_name == "production":
+            # For production method (multi-agent), no model parameter needed here - it's set dynamically
+            self.extraction_method = create_extraction_method(method_name)
+        elif method_name == "openai_structured":
+            # For OpenAI structured method, pass the model parameter
+            self.extraction_method = create_extraction_method(method_name, model=model)
+        else:
+            # For other methods, try to pass model parameter if supported
+            try:
+                self.extraction_method = create_extraction_method(method_name, model=model)
+            except TypeError:
+                # If method doesn't support model parameter, create without it
+                self.extraction_method = create_extraction_method(method_name)
+        
         self.requires_line_numbers = method_requires_line_numbers(method_name)
         self.requires_content_references = method_requires_content_references(method_name)
         

agentgraph/methods/production/openai_structured_extractor.py

@@ -342,6 +342,10 @@ class OpenAIStructuredFactory:
     def process_text(self, input_data: str) -> Dict[str, Any]:
         """Process text using the extractor."""
         return self.extractor.process_text(input_data)
+    
+    def extract_knowledge_graph(self, input_data: str, context_documents: Optional[List[Dict[str, Any]]] = None) -> KnowledgeGraph:
+        """Extract knowledge graph using the current model."""
+        return self.extractor.extract_knowledge_graph(input_data, context_documents)
 
 # Export factory instance
 openai_structured_factory = OpenAIStructuredFactory()