mirror of
https://github.com/Mintplex-Labs/langchain-python.git
synced 2026-07-25 04:26:41 -04:00
Leonid Ganeline
1c81883d42
This PR targets the `API Reference` documentation.
- Several classes and functions missed `docstrings`. These docstrings
were created.
- In several places this
```
except ImportError:
raise ValueError(
```
was replaced to
```
except ImportError:
raise ImportError(
```
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
from typing import Any, Dict
|
|
|
|
|
|
def _resolve_schema_references(schema: Any, definitions: Dict[str, Any]) -> Any:
|
|
"""
|
|
Resolves the $ref keys in a JSON schema object using the provided definitions.
|
|
"""
|
|
if isinstance(schema, list):
|
|
for i, item in enumerate(schema):
|
|
schema[i] = _resolve_schema_references(item, definitions)
|
|
elif isinstance(schema, dict):
|
|
if "$ref" in schema:
|
|
ref_key = schema.pop("$ref").split("/")[-1]
|
|
ref = definitions.get(ref_key, {})
|
|
schema.update(ref)
|
|
else:
|
|
for key, value in schema.items():
|
|
schema[key] = _resolve_schema_references(value, definitions)
|
|
return schema
|
|
|
|
|
|
def _convert_schema(schema: dict) -> dict:
|
|
props = {k: {"title": k, **v} for k, v in schema["properties"].items()}
|
|
return {
|
|
"type": "object",
|
|
"properties": props,
|
|
"required": schema.get("required", []),
|
|
}
|
|
|
|
|
|
def get_llm_kwargs(function: dict) -> dict:
|
|
"""Returns the kwargs for the LLMChain constructor.
|
|
|
|
Args:
|
|
function: The function to use.
|
|
|
|
Returns:
|
|
The kwargs for the LLMChain constructor.
|
|
"""
|
|
return {"functions": [function], "function_call": {"name": function["name"]}}
|