wu981526092 commited on
Commit
ab39c06
·
1 Parent(s): 5632471
agentgraph/methods/production/openai_structured_extractor.py CHANGED
@@ -330,7 +330,7 @@ def extract_knowledge_graph(input_data: str) -> KnowledgeGraph:
330
  class OpenAIStructuredFactory:
331
  """Factory class for OpenAI structured extraction method."""
332
 
333
- def __init__(self, model: str = "gpt-4o-2024-08-06"):
334
  self.model = model
335
  self.extractor = OpenAIStructuredExtractor(model)
336
 
 
330
  class OpenAIStructuredFactory:
331
  """Factory class for OpenAI structured extraction method."""
332
 
333
+ def __init__(self, model: str = "gpt-4o-mini"):
334
  self.model = model
335
  self.extractor = OpenAIStructuredExtractor(model)
336