nvidia-nat 1.3.0.dev2__py3-none-any.whl → 1.3.0rc2__py3-none-any.whl

nat/builder/function_base.py

@@ -111,7 +111,7 @@ class FunctionBase(typing.Generic[InputT, StreamingOutputT, SingleOutputT], ABC)
         ValueError
             If the input type cannot be determined from the class definition
         """
-        for base_cls in self.__class__.__orig_bases__:  # pylint: disable=no-member # type: ignore
+        for base_cls in self.__class__.__orig_bases__:
 
             base_cls_args = typing.get_args(base_cls)
 
@@ -196,7 +196,7 @@ class FunctionBase(typing.Generic[InputT, StreamingOutputT, SingleOutputT], ABC)
         ValueError
             If the streaming output type cannot be determined from the class definition
         """
-        for base_cls in self.__class__.__orig_bases__:  # pylint: disable=no-member # type: ignore
+        for base_cls in self.__class__.__orig_bases__:
 
             base_cls_args = typing.get_args(base_cls)
 
@@ -269,7 +269,7 @@ class FunctionBase(typing.Generic[InputT, StreamingOutputT, SingleOutputT], ABC)
         ValueError
             If the single output type cannot be determined from the class definition
         """
-        for base_cls in self.__class__.__orig_bases__:  # pylint: disable=no-member # type: ignore
+        for base_cls in self.__class__.__orig_bases__:
 
             base_cls_args = typing.get_args(base_cls)
 

nat/builder/function_info.py

@@ -231,9 +231,9 @@ class FunctionDescriptor:
         else:
             annotations = [param.annotation for param in sig.parameters.values()]
 
-            is_input_typed = all([a != sig.empty for a in annotations])  # pylint: disable=use-a-generator
+            is_input_typed = all([a != sig.empty for a in annotations])
 
-            input_type = tuple[*annotations] if is_input_typed else None  # noqa: syntax-error
+            input_type = tuple[*annotations] if is_input_typed else None
 
         # Get the base type here removing all annotations and async generators
         output_annotation_decomp = DecomposedType(sig.return_annotation).get_base_type()
@@ -372,18 +372,16 @@ class FunctionInfo:
 
             if (stream_to_single_fn is not None):
                 raise ValueError("Cannot provide both single_fn and stream_to_single_fn")
-        else:
-            if (stream_to_single_fn is not None and stream_fn is None):
-                raise ValueError("stream_fn must be provided if stream_to_single_fn is provided")
+        elif (stream_to_single_fn is not None and stream_fn is None):
+            raise ValueError("stream_fn must be provided if stream_to_single_fn is provided")
 
         if (stream_fn is not None):
             final_stream_fn = stream_fn
 
             if (single_to_stream_fn is not None):
                 raise ValueError("Cannot provide both stream_fn and single_to_stream_fn")
-        else:
-            if (single_to_stream_fn is not None and single_fn is None):
-                raise ValueError("single_fn must be provided if single_to_stream_fn is provided")
+        elif (single_to_stream_fn is not None and single_fn is None):
+            raise ValueError("single_fn must be provided if single_to_stream_fn is provided")
 
         if (single_fn is None and stream_fn is None):
             raise ValueError("At least one of single_fn or stream_fn must be provided")

nat/builder/user_interaction_manager.py

@@ -61,13 +61,13 @@ class UserInteractionManager:
 
         uuid_req = str(uuid.uuid4())
         status = InteractionStatus.IN_PROGRESS
-        timestamp = time.strftime("%Y-%m-%dT%H:%M:%SZ")
+        timestamp = time.strftime("%Y-%m-%dT%H:%M:%SZ", time.gmtime())
         sys_human_interaction = InteractionPrompt(id=uuid_req, status=status, timestamp=timestamp, content=content)
 
         resp = await self._context_state.user_input_callback.get()(sys_human_interaction)
 
         # Rebuild a InteractionResponse object with the response
-        timestamp = time.strftime("%Y-%m-%dT%H:%M:%SZ")
+        timestamp = time.strftime("%Y-%m-%dT%H:%M:%SZ", time.gmtime())
         status = InteractionStatus.COMPLETED
         sys_human_interaction = InteractionResponse(id=uuid_req, status=status, timestamp=timestamp, content=resp)
 

nat/builder/workflow.py

@@ -20,6 +20,7 @@ from typing import Any
 from nat.builder.context import ContextState
 from nat.builder.embedder import EmbedderProviderInfo
 from nat.builder.function import Function
+from nat.builder.function import FunctionGroup
 from nat.builder.function_base import FunctionBase
 from nat.builder.function_base import InputT
 from nat.builder.function_base import SingleOutputT
@@ -44,6 +45,7 @@ class Workflow(FunctionBase[InputT, StreamingOutputT, SingleOutputT]):
                  config: Config,
                  entry_fn: Function[InputT, StreamingOutputT, SingleOutputT],
                  functions: dict[str, Function] | None = None,
+                 function_groups: dict[str, FunctionGroup] | None = None,
                  llms: dict[str, LLMProviderInfo] | None = None,
                  embeddings: dict[str, EmbedderProviderInfo] | None = None,
                  memory: dict[str, MemoryEditor] | None = None,
@@ -59,6 +61,7 @@ class Workflow(FunctionBase[InputT, StreamingOutputT, SingleOutputT]):
 
         self.config = config
         self.functions = functions or {}
+        self.function_groups = function_groups or {}
         self.llms = llms or {}
         self.embeddings = embeddings or {}
         self.memory = memory or {}
@@ -83,6 +86,13 @@ class Workflow(FunctionBase[InputT, StreamingOutputT, SingleOutputT]):
 
         return self._entry_fn.has_single_output
 
+    async def get_all_exporters(self) -> dict[str, BaseExporter]:
+        return await self.exporter_manager.get_all_exporters()
+
+    @property
+    def exporter_manager(self) -> ExporterManager:
+        return self._exporter_manager.get()
+
     @asynccontextmanager
     async def run(self, message: InputT):
         """
@@ -93,7 +103,7 @@ class Workflow(FunctionBase[InputT, StreamingOutputT, SingleOutputT]):
         async with Runner(input_message=message,
                           entry_fn=self._entry_fn,
                           context_state=self._context_state,
-                          exporter_manager=self._exporter_manager.get()) as runner:
+                          exporter_manager=self.exporter_manager) as runner:
 
             # The caller can `yield runner` so they can do `runner.result()` or `runner.result_stream()`
             yield runner
@@ -119,6 +129,7 @@ class Workflow(FunctionBase[InputT, StreamingOutputT, SingleOutputT]):
                       config: Config,
                       entry_fn: Function[InputT, StreamingOutputT, SingleOutputT],
                       functions: dict[str, Function] | None = None,
+                      function_groups: dict[str, FunctionGroup] | None = None,
                       llms: dict[str, LLMProviderInfo] | None = None,
                       embeddings: dict[str, EmbedderProviderInfo] | None = None,
                       memory: dict[str, MemoryEditor] | None = None,
@@ -138,6 +149,7 @@ class Workflow(FunctionBase[InputT, StreamingOutputT, SingleOutputT]):
         return WorkflowImpl(config=config,
                             entry_fn=entry_fn,
                             functions=functions,
+                            function_groups=function_groups,
                             llms=llms,
                             embeddings=embeddings,
                             memory=memory,