참고소스 수정본
This commit is contained in:
@@ -0,0 +1 @@
|
||||
"""Provider implementation."""
|
||||
92
참고/instructor-main/instructor/providers/gemini/client.py
Normal file
92
참고/instructor-main/instructor/providers/gemini/client.py
Normal file
@@ -0,0 +1,92 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Any, Literal, overload
|
||||
|
||||
import google.generativeai as genai # type: ignore[import-not-found]
|
||||
|
||||
import instructor
|
||||
|
||||
|
||||
@overload
|
||||
def from_gemini(
|
||||
client: genai.GenerativeModel,
|
||||
mode: instructor.Mode = instructor.Mode.GEMINI_JSON,
|
||||
use_async: Literal[True] = True,
|
||||
**kwargs: Any,
|
||||
) -> instructor.AsyncInstructor: ...
|
||||
|
||||
|
||||
@overload
|
||||
def from_gemini(
|
||||
client: genai.GenerativeModel,
|
||||
mode: instructor.Mode = instructor.Mode.GEMINI_JSON,
|
||||
use_async: Literal[False] = False,
|
||||
**kwargs: Any,
|
||||
) -> instructor.Instructor: ...
|
||||
|
||||
|
||||
def from_gemini(
|
||||
client: genai.GenerativeModel,
|
||||
mode: instructor.Mode = instructor.Mode.GEMINI_JSON,
|
||||
use_async: bool = False,
|
||||
**kwargs: Any,
|
||||
) -> instructor.Instructor | instructor.AsyncInstructor:
|
||||
import warnings
|
||||
|
||||
warnings.warn(
|
||||
"from_gemini is deprecated and will be removed in a future version. "
|
||||
"Please use from_genai or from_provider instead. "
|
||||
"Install google-genai with: pip install google-genai\n"
|
||||
"Example migration:\n"
|
||||
" # Old way\n"
|
||||
" from instructor import from_gemini\n"
|
||||
" import google.generativeai as genai\n"
|
||||
" client = from_gemini(genai.GenerativeModel('gemini-3-flash'))\n\n"
|
||||
" # New way\n"
|
||||
" from instructor import from_genai\n"
|
||||
" from google import genai\n"
|
||||
" client = from_genai(genai.Client())\n"
|
||||
" # OR use from_provider\n"
|
||||
" client = instructor.from_provider('google/gemini-3-flash')",
|
||||
DeprecationWarning,
|
||||
stacklevel=2,
|
||||
)
|
||||
|
||||
valid_modes = {
|
||||
instructor.Mode.GEMINI_JSON,
|
||||
instructor.Mode.GEMINI_TOOLS,
|
||||
}
|
||||
|
||||
if mode not in valid_modes:
|
||||
from ...core.exceptions import ModeError
|
||||
|
||||
raise ModeError(
|
||||
mode=str(mode), provider="Gemini", valid_modes=[str(m) for m in valid_modes]
|
||||
)
|
||||
|
||||
if not isinstance(client, genai.GenerativeModel):
|
||||
from ...core.exceptions import ClientError
|
||||
|
||||
raise ClientError(
|
||||
f"Client must be an instance of genai.GenerativeModel. "
|
||||
f"Got: {type(client).__name__}"
|
||||
)
|
||||
|
||||
if use_async:
|
||||
create = client.generate_content_async
|
||||
return instructor.AsyncInstructor(
|
||||
client=client,
|
||||
create=instructor.patch(create=create, mode=mode),
|
||||
provider=instructor.Provider.GEMINI,
|
||||
mode=mode,
|
||||
**kwargs,
|
||||
)
|
||||
|
||||
create = client.generate_content
|
||||
return instructor.Instructor(
|
||||
client=client,
|
||||
create=instructor.patch(create=create, mode=mode),
|
||||
provider=instructor.Provider.GEMINI,
|
||||
mode=mode,
|
||||
**kwargs,
|
||||
)
|
||||
1229
참고/instructor-main/instructor/providers/gemini/utils.py
Normal file
1229
참고/instructor-main/instructor/providers/gemini/utils.py
Normal file
File diff suppressed because it is too large
Load Diff
Reference in New Issue
Block a user