fix typing

run-llama · Oct 8, 2024 · 5ad2924 · 5ad2924
1 parent 5ef6381
commit 5ad2924
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/templates/components/engines/python/agent/tools/artifact.py b/templates/components/engines/python/agent/tools/artifact.py
@@ -87,7 +87,7 @@ def artifact(self, query: str, old_code: Optional[str] = None) -> Dict:
             ChatMessage(role="user", content=user_message),
         ]
         try:
-            sllm = Settings.llm.as_structured_llm(output_cls=CodeArtifact)
+            sllm = Settings.llm.as_structured_llm(output_cls=CodeArtifact)  # type: ignore
             response = sllm.chat(messages)
             data: CodeArtifact = response.raw
             return data.model_dump()

diff --git a/templates/types/streaming/fastapi/app/engine/__init__.py b/templates/types/streaming/fastapi/app/engine/__init__.py
@@ -1 +1,3 @@
 from .engine import get_chat_engine as get_chat_engine
+
+__all__ = ["get_chat_engine"]
Original file line number	Diff line number	Diff line change
		@@ -1 +1,3 @@
		from .engine import get_chat_engine as get_chat_engine

		__all__ = ["get_chat_engine"]