More poetry run fixes for the pipeline and bring back litellm to matc…

…h lock file, also because newer versions still have the openai stream=True not working bug
rogeriochaves · Sep 27, 2023 · 4565d9e · 4565d9e
1 parent 874be7f
commit 4565d9e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Makefile b/Makefile
@@ -7,7 +7,7 @@ test-integration:
 	PYTHONPATH=$$PYTHONPATH:. poetry run pytest -s -m integration $(filter-out $@,$(MAKECMDGOALS))
 
 doctest:
-	PYTHONPATH=$$PYTHONPATH:. poetry run pytest --doctest-modules langstream/utils && PYTHONPATH=$PYTHONPATH:. pytest --doctest-modules langstream/core && PYTHONPATH=$PYTHONPATH:. pytest --doctest-modules langstream/contrib/llms
+	PYTHONPATH=$$PYTHONPATH:. poetry run pytest --doctest-modules langstream/utils && PYTHONPATH=$PYTHONPATH:. poetry run pytest --doctest-modules langstream/core && PYTHONPATH=$PYTHONPATH:. poetry run pytest --doctest-modules langstream/contrib/llms
 
 nbtest:
 	poetry run nbdoc_test --fname docs/docs/
@@ -16,7 +16,7 @@ docs:
 	make pdocs && make nbdocs && cd docs && npm run build
 
 pdocs:
-	pdoc --html -o ./docs/static/reference --template-dir ./docs/pdoc_template langstream --force
+	poetry run pdoc --html -o ./docs/static/reference --template-dir ./docs/pdoc_template langstream --force
 
 nbdocs:
 	poetry run nbdoc_build --srcdir docs/docs

diff --git a/pyproject.toml b/pyproject.toml
@@ -33,7 +33,7 @@ retry = { version = "*" }
 
 openai = { version = "*", optional = true }
 gpt4all = { version = "*", optional = true }
-litellm = { version = "*", optional = true }
+litellm = { version = "=0.1.450", optional = true }
 
 [tool.poetry.group.dev.dependencies]
 pdoc3 = "^0.10.0"