Enable Import Sorting

.ruff.toml

@@ -31,6 +31,8 @@ target-version = "py39"
 
 [lint]
 select = [
+  # isort
+  "I",
   # bugbear rules
   "B",
   # remove unused imports

examples/anthropic/run.py

@@ -1,5 +1,6 @@
-from pydantic import BaseModel
 import anthropic
+from pydantic import BaseModel
+
 import instructor
 
 # Patching the Anthropics client with the instructor for enhanced capabilities

examples/auto-ticketer/run.py

@@ -1,9 +1,10 @@
-import instructor
-from openai import OpenAI
-
+from enum import Enum
 from typing import Optional
+
+from openai import OpenAI
 from pydantic import BaseModel, Field
-from enum import Enum
+
+import instructor
 
 client = instructor.from_openai(OpenAI())
 

examples/avail/run.py

@@ -1,9 +1,10 @@
-from pydantic import BaseModel, Field
-from typing import Literal
 from collections.abc import Iterable
 from datetime import datetime, timedelta
+from typing import Literal
 
 from openai import OpenAI
+from pydantic import BaseModel, Field
+
 import instructor
 
 client = instructor.from_openai(OpenAI())

examples/avail/run_mixtral.py

@@ -1,9 +1,10 @@
 import os
-from pydantic import BaseModel, Field
-from typing import Literal
 from datetime import datetime, timedelta
+from typing import Literal
 
 from openai import OpenAI
+from pydantic import BaseModel, Field
+
 import instructor
 
 client = instructor.from_openai(

examples/batch-classification/run-cache.py

@@ -1,9 +1,10 @@
-import instructor
 import asyncio
+from enum import Enum
 
 from openai import AsyncOpenAI
 from pydantic import BaseModel, Field, field_validator
-from enum import Enum
+
+import instructor
 
 client = instructor.from_openai(AsyncOpenAI(), mode=instructor.Mode.TOOLS)
 sem = asyncio.Semaphore(5)

examples/batch-classification/run.py

@@ -1,10 +1,11 @@
-import json
-import instructor
 import asyncio
+import json
+from enum import Enum
 
 from openai import AsyncOpenAI
 from pydantic import BaseModel, Field, field_validator
-from enum import Enum
+
+import instructor
 
 client = AsyncOpenAI()
 client = instructor.from_openai(client, mode=instructor.Mode.TOOLS)

examples/batch-classification/run_langsmith.py

@@ -1,12 +1,12 @@
-import instructor
 import asyncio
+from enum import Enum
 
 from langsmith import traceable
 from langsmith.wrappers import wrap_openai
-
 from openai import AsyncOpenAI
 from pydantic import BaseModel, Field, field_validator
-from enum import Enum
+
+import instructor
 
 client = wrap_openai(AsyncOpenAI())
 client = instructor.from_openai(client, mode=instructor.Mode.TOOLS)

examples/caching/example_diskcache.py

@@ -1,11 +1,12 @@
 import functools
 import inspect
-import instructor
-import diskcache
 
-from openai import OpenAI, AsyncOpenAI
+import diskcache
+from openai import AsyncOpenAI, OpenAI
 from pydantic import BaseModel
 
+import instructor
+
 client = instructor.from_openai(OpenAI())
 aclient = instructor.from_openai(AsyncOpenAI())
 

examples/caching/example_redis.py

@@ -1,10 +1,11 @@
-import redis
 import functools
 import inspect
-import instructor
 
-from pydantic import BaseModel
+import redis
 from openai import OpenAI
+from pydantic import BaseModel
+
+import instructor
 
 client = instructor.from_openai(OpenAI())
 cache = redis.Redis("localhost")

examples/caching/lru.py

@@ -1,7 +1,9 @@
-import instructor
+import functools
+
 from openai import OpenAI
 from pydantic import BaseModel
-import functools
+
+import instructor
 
 client = instructor.from_openai(OpenAI())
 

examples/chain-of-density/chain_of_density.py

@@ -1,8 +1,9 @@
-from pydantic import BaseModel, Field, field_validator
-import instructor
 import nltk
-from openai import OpenAI
 import spacy
+from openai import OpenAI
+from pydantic import BaseModel, Field, field_validator
+
+import instructor
 
 client = instructor.from_openai(OpenAI())
 nlp = spacy.load("en_core_web_sm")

examples/chain-of-density/finetune.py

@@ -1,10 +1,12 @@
-from openai import OpenAI
-from chain_of_density import summarize_article
 import csv
 import logging
-import instructor
+
+from chain_of_density import summarize_article
+from openai import OpenAI
 from pydantic import BaseModel, Field
 
+import instructor
+
 logging.basicConfig(level=logging.INFO)
 
 client = instructor.from_openai(OpenAI())

examples/citation_with_extraction/citation_fuzzy_match.py

@@ -1,8 +1,8 @@
-import instructor
-
 from loguru import logger
 from openai import OpenAI
-from pydantic import Field, BaseModel, FieldValidationInfo, model_validator
+from pydantic import BaseModel, Field, FieldValidationInfo, model_validator
+
+import instructor
 
 client = instructor.from_openai(OpenAI())
 

examples/citation_with_extraction/diagram.py

@@ -1,5 +1,4 @@
 import erdantic as erd
-
 from citation_fuzzy_match import QuestionAnswer
 
 diagram = erd.create(QuestionAnswer)

examples/citation_with_extraction/main.py

@@ -1,16 +1,16 @@
 import json
+import logging
+import os
 from collections.abc import Iterable
-from fastapi import FastAPI, Request, HTTPException
+
+from fastapi import FastAPI, HTTPException, Request
 from fastapi.params import Depends
-from instructor import OpenAISchema
+from openai import OpenAI
 from pydantic import BaseModel, Field
 from starlette.responses import StreamingResponse
 
-import os
 import instructor
-import logging
-
-from openai import OpenAI
+from instructor import OpenAISchema
 from instructor.dsl.multitask import MultiTaskBase
 
 client = instructor.from_openai(OpenAI())

examples/citation_with_extraction/modal_main.py

@@ -1,5 +1,5 @@
-from main import app
 import modal
+from main import app
 
 stub = modal.Stub("rag-citation")
 

examples/citations/run.py

@@ -1,4 +1,5 @@
 from typing import Optional
+
 from openai import OpenAI
 from pydantic import (
     BaseModel,

examples/classification/classifiy_with_validation.py

@@ -1,9 +1,10 @@
 # pip install openai instructor
-from pydantic import BaseModel, field_validator, Field
 import openai
-import instructor
+from pydantic import BaseModel, Field, field_validator
 from tqdm import tqdm
 
+import instructor
+
 client = instructor.from_openai(openai.OpenAI())
 
 classes = {

examples/classification/multi_prediction.py

@@ -1,9 +1,10 @@
 import enum
-import instructor
 
 from openai import OpenAI
 from pydantic import BaseModel
 
+import instructor
+
 client = instructor.from_openai(OpenAI())
 
 

examples/classification/simple_prediction.py

@@ -1,9 +1,10 @@
 import enum
-import instructor
-from openai import OpenAI
 
+from openai import OpenAI
 from pydantic import BaseModel
 
+import instructor
+
 client = instructor.from_openai(OpenAI())
 
 

examples/codegen-from-schema/create_fastapi_app.py

@@ -1,9 +1,10 @@
-import json
 import datetime
-from pathlib import Path
-from jinja2 import Template
+import json
 import re
+from pathlib import Path
+
 from datamodel_code_generator import InputFileType, generate
+from jinja2 import Template
 from pydantic import BaseModel
 
 APP_TEMPLATE_STR = '''# generated by instructor-codegen:

examples/codegen-from-schema/run.py

@@ -4,13 +4,13 @@
 #   api_path: /api/v1/extract_person
 #   json_schema_path: ./input.json
 
-import instructor
-
 from fastapi import FastAPI
-from pydantic import BaseModel
 from jinja2 import Template
 from models import ExtractPerson
 from openai import AsyncOpenAI
+from pydantic import BaseModel
+
+import instructor
 
 aclient = instructor.apatch(AsyncOpenAI())
 

examples/cohere/cohere.py

@@ -1,7 +1,7 @@
 import cohere
-import instructor
 from pydantic import BaseModel, Field
 
+import instructor
 
 # Patching the Cohere client with the instructor for enhanced capabilities
 client = instructor.from_cohere(

examples/crm/run.py

@@ -1,7 +1,9 @@
 from enum import Enum
+
+from openai import OpenAI
 from pydantic import BaseModel, Field
+
 import instructor
-from openai import OpenAI
 
 client = instructor.from_openai(OpenAI())
 

examples/distilations/three_digit_mul.py

@@ -1,6 +1,7 @@
 import logging
 
 from pydantic import BaseModel, Field
+
 from instructor import Instructions
 
 logging.basicConfig(level=logging.INFO)

examples/distilations/three_digit_mul_dispatch.py

@@ -1,8 +1,9 @@
 import logging
 
 from pydantic import BaseModel, Field
-from instructor import Instructions
+
 import instructor
+from instructor import Instructions
 
 instructor.from_openai()
 

examples/evals/eval.py

@@ -1,11 +1,12 @@
+import json
 from collections import Counter, defaultdict
 from enum import Enum
+from pprint import pprint
 from typing import Any, Union
+
+import models as m
 import numpy as np
-import json
 from pydantic import ValidationError
-from pprint import pprint
-import models as m
 
 
 class Status(Enum):

examples/evals/models.py

@@ -1,6 +1,7 @@
+from enum import Enum
 from typing import Optional
+
 from pydantic import BaseModel, Field
-from enum import Enum
 
 
 class SourceType(str, Enum):

examples/extract-table/run_vision.py

@@ -1,17 +1,19 @@
-from openai import OpenAI
 from io import StringIO
 from typing import Annotated, Any
+
+import pandas as pd
+from openai import OpenAI
 from pydantic import (
     BaseModel,
     BeforeValidator,
-    PlainSerializer,
     InstanceOf,
+    PlainSerializer,
     WithJsonSchema,
 )
-import instructor
-import pandas as pd
 from rich.console import Console
 
+import instructor
+
 console = Console()
 client = instructor.from_openai(
     client=OpenAI(),