mirror of
https://github.com/hwchase17/langchain.git
synced 2025-06-25 08:03:39 +00:00
Core: google docstring parsing fix (#28404)
Thank you for contributing to LangChain! - [ ] **PR title**: "core: google docstring parsing fix" - [x] **PR message**: - **Description:** Added a solution for invalid parsing of google docstring such as: Args: net_annual_income (float): The user's net annual income (in current year dollars). - **Issue:** Previous code would return arg = "net_annual_income (float)" which would cause exception in _validate_docstring_args_against_annotations - **Dependencies:** None If no one reviews your PR within a few days, please @-mention one of baskaryan, efriis, eyurtsev, ccurme, vbarda, hwchase17. Co-authored-by: Erick Friis <erick@langchain.dev>
This commit is contained in:
parent
b78b2f7a28
commit
4e743b5427
@ -646,9 +646,13 @@ def _parse_google_docstring(
|
||||
for line in args_block.split("\n")[1:]:
|
||||
if ":" in line:
|
||||
arg, desc = line.split(":", maxsplit=1)
|
||||
arg_descriptions[arg.strip()] = desc.strip()
|
||||
arg = arg.strip()
|
||||
arg_name, _, _annotations = arg.partition(" ")
|
||||
if _annotations.startswith("(") and _annotations.endswith(")"):
|
||||
arg = arg_name
|
||||
arg_descriptions[arg] = desc.strip()
|
||||
elif arg:
|
||||
arg_descriptions[arg.strip()] += " " + line.strip()
|
||||
arg_descriptions[arg] += " " + line.strip()
|
||||
return description, arg_descriptions
|
||||
|
||||
|
||||
|
@ -71,6 +71,19 @@ def function() -> Callable:
|
||||
return dummy_function
|
||||
|
||||
|
||||
@pytest.fixture()
|
||||
def function_docstring_annotations() -> Callable:
|
||||
def dummy_function(arg1: int, arg2: Literal["bar", "baz"]) -> None:
|
||||
"""dummy function
|
||||
|
||||
Args:
|
||||
arg1 (int): foo
|
||||
arg2: one of 'bar', 'baz'
|
||||
"""
|
||||
|
||||
return dummy_function
|
||||
|
||||
|
||||
@pytest.fixture()
|
||||
def runnable() -> Runnable:
|
||||
class Args(ExtensionsTypedDict):
|
||||
@ -278,6 +291,7 @@ class DummyWithClassMethod:
|
||||
def test_convert_to_openai_function(
|
||||
pydantic: type[BaseModel],
|
||||
function: Callable,
|
||||
function_docstring_annotations: Callable,
|
||||
dummy_structured_tool: StructuredTool,
|
||||
dummy_tool: BaseTool,
|
||||
json_schema: dict,
|
||||
@ -311,6 +325,7 @@ def test_convert_to_openai_function(
|
||||
for fn in (
|
||||
pydantic,
|
||||
function,
|
||||
function_docstring_annotations,
|
||||
dummy_structured_tool,
|
||||
dummy_tool,
|
||||
json_schema,
|
||||
|
Loading…
Reference in New Issue
Block a user