wu981526092 commited on
Commit
f0fc928
·
1 Parent(s): b522e95
agentgraph/shared/method_registry.py CHANGED
@@ -23,6 +23,18 @@ class SchemaType(Enum):
23
 
24
  # Method Registry - maps method names to their implementations and schemas
25
  AVAILABLE_METHODS = {
 
 
 
 
 
 
 
 
 
 
 
 
26
  # Production method using reference-based schema
27
  "production": {
28
  "name": "Multi-Agent Knowledge Extractor",
@@ -150,7 +162,7 @@ AVAILABLE_METHODS = {
150
 
151
 
152
  # Default method configuration
153
- DEFAULT_METHOD = "production"
154
 
155
 
156
  def get_available_methods() -> Dict[str, Dict[str, Any]]:
@@ -226,4 +238,4 @@ def get_method_description(method_name: str) -> str:
226
  def validate_method_schema_compatibility(method_name: str, expected_schema: SchemaType) -> bool:
227
  """Validate that a method uses the expected schema type"""
228
  method_schema = get_schema_for_method(method_name)
229
- return method_schema == expected_schema if method_schema else False
 
23
 
24
  # Method Registry - maps method names to their implementations and schemas
25
  AVAILABLE_METHODS = {
26
+ # Production method: OpenAI Structured Outputs (simple, direct_call)
27
+ "openai_structured": {
28
+ "name": "OpenAI Structured Outputs",
29
+ "description": "Simple OpenAI structured outputs extractor using Pydantic models",
30
+ "method_type": MethodType.PRODUCTION,
31
+ "schema_type": SchemaType.REFERENCE_BASED,
32
+ "module_path": "agentgraph.methods.production.openai_structured_extractor",
33
+ "class_name": "OpenAIStructuredFactory",
34
+ "supported_features": ["structured_outputs", "direct_extraction"],
35
+ "processing_type": "direct_call"
36
+ },
37
+
38
  # Production method using reference-based schema
39
  "production": {
40
  "name": "Multi-Agent Knowledge Extractor",
 
162
 
163
 
164
  # Default method configuration
165
+ DEFAULT_METHOD = "openai_structured"
166
 
167
 
168
  def get_available_methods() -> Dict[str, Dict[str, Any]]:
 
238
  def validate_method_schema_compatibility(method_name: str, expected_schema: SchemaType) -> bool:
239
  """Validate that a method uses the expected schema type"""
240
  method_schema = get_schema_for_method(method_name)
241
+ return method_schema == expected_schema if method_schema else False