mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-09 13:00:34 +00:00
a0c2281540
```python
"""python scripts/update_mypy_ruff.py"""
import glob
import tomllib
from pathlib import Path
import toml
import subprocess
import re
ROOT_DIR = Path(__file__).parents[1]
def main():
for path in glob.glob(str(ROOT_DIR / "libs/**/pyproject.toml"), recursive=True):
print(path)
with open(path, "rb") as f:
pyproject = tomllib.load(f)
try:
pyproject["tool"]["poetry"]["group"]["typing"]["dependencies"]["mypy"] = (
"^1.10"
)
pyproject["tool"]["poetry"]["group"]["lint"]["dependencies"]["ruff"] = (
"^0.5"
)
except KeyError:
continue
with open(path, "w") as f:
toml.dump(pyproject, f)
cwd = "/".join(path.split("/")[:-1])
completed = subprocess.run(
"poetry lock --no-update; poetry install --with typing; poetry run mypy . --no-color",
cwd=cwd,
shell=True,
capture_output=True,
text=True,
)
logs = completed.stdout.split("\n")
to_ignore = {}
for l in logs:
if re.match("^(.*)\:(\d+)\: error:.*\[(.*)\]", l):
path, line_no, error_type = re.match(
"^(.*)\:(\d+)\: error:.*\[(.*)\]", l
).groups()
if (path, line_no) in to_ignore:
to_ignore[(path, line_no)].append(error_type)
else:
to_ignore[(path, line_no)] = [error_type]
print(len(to_ignore))
for (error_path, line_no), error_types in to_ignore.items():
all_errors = ", ".join(error_types)
full_path = f"{cwd}/{error_path}"
try:
with open(full_path, "r") as f:
file_lines = f.readlines()
except FileNotFoundError:
continue
file_lines[int(line_no) - 1] = (
file_lines[int(line_no) - 1][:-1] + f" # type: ignore[{all_errors}]\n"
)
with open(full_path, "w") as f:
f.write("".join(file_lines))
subprocess.run(
"poetry run ruff format .; poetry run ruff --select I --fix .",
cwd=cwd,
shell=True,
capture_output=True,
text=True,
)
if __name__ == "__main__":
main()
```
44 lines
2.2 KiB
Python
44 lines
2.2 KiB
Python
"""Test Nebula API wrapper."""
|
||
|
||
from langchain_community.llms.symblai_nebula import Nebula
|
||
|
||
|
||
def test_symblai_nebula_call() -> None:
|
||
"""Test valid call to Nebula."""
|
||
conversation = """Sam: Good morning, team! Let's keep this standup concise.
|
||
We'll go in the usual order: what you did yesterday,
|
||
what you plan to do today, and any blockers. Alex, kick us off.
|
||
Alex: Morning! Yesterday, I wrapped up the UI for the user dashboard.
|
||
The new charts and widgets are now responsive.
|
||
I also had a sync with the design team to ensure the final touchups are in
|
||
line with the brand guidelines. Today, I'll start integrating the frontend with
|
||
the new API endpoints Rhea was working on.
|
||
The only blocker is waiting for some final API documentation,
|
||
but I guess Rhea can update on that.
|
||
Rhea: Hey, all! Yep, about the API documentation - I completed the majority of
|
||
the backend work for user data retrieval yesterday.
|
||
The endpoints are mostly set up, but I need to do a bit more testing today.
|
||
I'll finalize the API documentation by noon, so that should unblock Alex.
|
||
After that, I’ll be working on optimizing the database queries
|
||
for faster data fetching. No other blockers on my end.
|
||
Sam: Great, thanks Rhea. Do reach out if you need any testing assistance
|
||
or if there are any hitches with the database.
|
||
Now, my update: Yesterday, I coordinated with the client to get clarity
|
||
on some feature requirements. Today, I'll be updating our project roadmap
|
||
and timelines based on their feedback. Additionally, I'll be sitting with
|
||
the QA team in the afternoon for preliminary testing.
|
||
Blocker: I might need both of you to be available for a quick call
|
||
in case the client wants to discuss the changes live.
|
||
Alex: Sounds good, Sam. Just let us know a little in advance for the call.
|
||
Rhea: Agreed. We can make time for that.
|
||
Sam: Perfect! Let's keep the momentum going. Reach out if there are any
|
||
sudden issues or support needed. Have a productive day!
|
||
Alex: You too.
|
||
Rhea: Thanks, bye!"""
|
||
llm = Nebula(nebula_api_key="<your_api_key>") # type: ignore[arg-type]
|
||
|
||
instruction = """Identify the main objectives mentioned in this
|
||
conversation."""
|
||
output = llm.invoke(f"{instruction}\n{conversation}")
|
||
assert isinstance(output, str)
|