Explicitly specify framework_wrappers for langchain tools

Signed-off-by: David Gardner <dagardner@nvidia.com>

Author: dagardner-nv

packages/nvidia_nat_langchain/src/nat/plugins/langchain/tools/code_generation_tool.py

@@ -36,7 +36,7 @@ class CodeGenerationTool(FunctionBaseConfig, name="code_generation"):
                         "this tool!")
 
 
-@register_function(config_type=CodeGenerationTool)
+@register_function(config_type=CodeGenerationTool, framework_wrappers=[LLMFrameworkEnum.LANGCHAIN])
 async def code_generation_tool(config: CodeGenerationTool, builder: Builder):
     from langchain_core.prompts.chat import ChatPromptTemplate
 

packages/nvidia_nat_langchain/src/nat/plugins/langchain/tools/tavily_internet_search.py

@@ -17,6 +17,7 @@
 from pydantic import Field
 
 from nat.builder.builder import Builder
+from nat.builder.framework_enum import LLMFrameworkEnum
 from nat.builder.function_info import FunctionInfo
 from nat.cli.register_workflow import register_function
 from nat.data_models.common import SerializableSecretStr
@@ -39,7 +40,7 @@ class TavilyInternetSearchToolConfig(FunctionBaseConfig, name="tavily_internet_s
         description="Depth for relevance vs latency tradeoff - 'basic', 'advanced', 'fast', or 'ultra-fast'")
 
 
-@register_function(config_type=TavilyInternetSearchToolConfig)
+@register_function(config_type=TavilyInternetSearchToolConfig, framework_wrappers=[LLMFrameworkEnum.LANGCHAIN])
 async def tavily_internet_search(tool_config: TavilyInternetSearchToolConfig, builder: Builder):
     import os
 

packages/nvidia_nat_langchain/src/nat/plugins/langchain/tools/wikipedia_search.py

@@ -14,6 +14,7 @@
 # limitations under the License.
 
 from nat.builder.builder import Builder
+from nat.builder.framework_enum import LLMFrameworkEnum
 from nat.builder.function_info import FunctionInfo
 from nat.cli.register_workflow import register_function
 from nat.data_models.function import FunctionBaseConfig
@@ -28,7 +29,7 @@ class WikiSearchToolConfig(FunctionBaseConfig, name="wiki_search"):
 
 
 # Wiki search
-@register_function(config_type=WikiSearchToolConfig)
+@register_function(config_type=WikiSearchToolConfig, framework_wrappers=[LLMFrameworkEnum.LANGCHAIN])
 async def wiki_search(tool_config: WikiSearchToolConfig, builder: Builder):
     from langchain_community.document_loaders import WikipediaLoader