Skip to content

Commit

Permalink
chore: Clean up test files and consolidate vertex documentation
Browse files Browse the repository at this point in the history
  • Loading branch information
devin-ai-integration[bot] committed Nov 18, 2024
1 parent 22c1ddb commit 093e213
Show file tree
Hide file tree
Showing 2 changed files with 127 additions and 257 deletions.
127 changes: 127 additions & 0 deletions docs/examples/recursive.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,127 @@
---
title: Working with Recursive Schemas in Instructor
description: Learn how to effectively implement and use recursive Pydantic models for handling nested and hierarchical data structures.
---

# Recursive Schema Implementation Guide

This guide demonstrates how to work with recursive schemas in Instructor using Pydantic models. While flat schemas are often simpler to work with, some use cases require recursive structures to represent hierarchical data effectively.

!!! tips "Motivation"
Recursive schemas are particularly useful when dealing with:
* Nested organizational structures
* File system hierarchies
* Comment threads with replies
* Task dependencies with subtasks
* Abstract syntax trees

## Defining a Recursive Schema

Here's an example of how to define a recursive Pydantic model:

```python
from typing import List, Optional
from pydantic import BaseModel, Field

class RecursiveNode(BaseModel):
"""A node that can contain child nodes of the same type."""

name: str = Field(..., description="Name of the node")
value: Optional[str] = Field(None, description="Optional value associated with the node")
children: List["RecursiveNode"] = Field(
default_factory=list,
description="List of child nodes"
)

# Required for recursive Pydantic models
RecursiveNode.model_rebuild()
```

## Example Usage

Let's see how to use this recursive schema with Instructor:

```python
import instructor
from openai import OpenAI

client = instructor.from_openai(OpenAI())

def parse_hierarchy(text: str) -> RecursiveNode:
"""Parse text into a hierarchical structure."""
return client.chat.completions.create(
model="gpt-4",
messages=[
{
"role": "system",
"content": "You are an expert at parsing text into hierarchical structures."
},
{
"role": "user",
"content": f"Parse this text into a hierarchical structure: {text}"
}
],
response_model=RecursiveNode
)

# Example usage
hierarchy = parse_hierarchy("""
Company: Acme Corp
- Department: Engineering
- Team: Frontend
- Project: Website Redesign
- Project: Mobile App
- Team: Backend
- Project: API v2
- Project: Database Migration
- Department: Marketing
- Team: Digital
- Project: Social Media Campaign
- Team: Brand
- Project: Logo Refresh
""")
```

## Validation and Best Practices

When working with recursive schemas:

1. Always call `model_rebuild()` after defining the model
2. Consider adding validation for maximum depth to prevent infinite recursion
3. Use type hints properly to maintain code clarity
4. Consider implementing custom validators for specific business rules

```python
from pydantic import model_validator

class RecursiveNodeWithDepth(RecursiveNode):
@model_validator(mode='after')
def validate_depth(self) -> "RecursiveNodeWithDepth":
def check_depth(node: "RecursiveNodeWithDepth", current_depth: int = 0) -> int:
if current_depth > 10: # Maximum allowed depth
raise ValueError("Maximum depth exceeded")
return max(
[check_depth(child, current_depth + 1) for child in node.children],
default=current_depth
)

check_depth(self)
return self
```

## Performance Considerations

While recursive schemas are powerful, they can be more challenging for language models to handle correctly. Consider these tips:

1. Keep structures as shallow as possible
2. Use clear naming conventions
3. Provide good examples in your prompts
4. Consider breaking very large structures into smaller chunks

## Conclusion

Recursive schemas provide a powerful way to handle hierarchical data structures in your applications. While they require more careful handling than flat schemas, they can be invaluable for certain use cases.

For more examples of working with complex data structures, check out:
1. [Query Planning with Dependencies](planning-tasks.md)
2. [Knowledge Graph Generation](knowledge_graph.md)
257 changes: 0 additions & 257 deletions docs/integrations/vertexai.md

This file was deleted.

0 comments on commit 093e213

Please sign in to comment.