diff options
| author | Shreyans Jain <[email protected]> | 2025-10-08 15:54:42 -0700 |
|---|---|---|
| committer | GitHub <[email protected]> | 2025-10-08 15:54:42 -0700 |
| commit | 4467b65524c4958bfe5f76d36fc50e8e4e31c731 (patch) | |
| tree | 28ddef94669bea50e6fe86b36bec7ba4a99f29cf /packages/openai-sdk-python/src | |
| parent | feat: manual mcp and quick click to open (#464) (diff) | |
| download | supermemory-4467b65524c4958bfe5f76d36fc50e8e4e31c731.tar.xz supermemory-4467b65524c4958bfe5f76d36fc50e8e4e31c731.zip | |
fix: openai python sdk package export (#432)
Diffstat (limited to 'packages/openai-sdk-python/src')
| -rw-r--r-- | packages/openai-sdk-python/src/supermemory_openai/__init__.py (renamed from packages/openai-sdk-python/src/__init__.py) | 2 | ||||
| -rw-r--r-- | packages/openai-sdk-python/src/supermemory_openai/tools.py (renamed from packages/openai-sdk-python/src/tools.py) | 8 |
2 files changed, 4 insertions, 6 deletions
diff --git a/packages/openai-sdk-python/src/__init__.py b/packages/openai-sdk-python/src/supermemory_openai/__init__.py index b8564471..f05816e0 100644 --- a/packages/openai-sdk-python/src/__init__.py +++ b/packages/openai-sdk-python/src/supermemory_openai/__init__.py @@ -16,8 +16,6 @@ from .tools import ( create_add_memory_tool, ) -__version__ = "0.1.0" - __all__ = [ # Tools "SupermemoryTools", diff --git a/packages/openai-sdk-python/src/tools.py b/packages/openai-sdk-python/src/supermemory_openai/tools.py index 6dfe3d2f..ec87db10 100644 --- a/packages/openai-sdk-python/src/tools.py +++ b/packages/openai-sdk-python/src/supermemory_openai/tools.py @@ -6,7 +6,7 @@ from typing import Dict, List, Optional, Union, TypedDict from openai.types.chat import ( ChatCompletionMessageToolCall, ChatCompletionToolMessageParam, - ChatCompletionToolParam, + ChatCompletionFunctionToolParam, ) import supermemory from supermemory.types import ( @@ -50,7 +50,7 @@ class MemoryAddResult(TypedDict, total=False): # Function schemas for OpenAI function calling -MEMORY_TOOL_SCHEMAS = { +MEMORY_TOOL_SCHEMAS: Dict[str, ChatCompletionFunctionToolParam] = { "search_memories": { "name": "search_memories", "description": ( @@ -129,7 +129,7 @@ class SupermemoryTools: else: self.container_tags = ["sm_project_default"] - def get_tool_definitions(self) -> List[ChatCompletionToolParam]: + def get_tool_definitions(self) -> List[ChatCompletionFunctionToolParam]: """Get OpenAI function definitions for all memory tools. Returns: @@ -247,7 +247,7 @@ def create_supermemory_tools( return SupermemoryTools(api_key, config) -def get_memory_tool_definitions() -> List[ChatCompletionToolParam]: +def get_memory_tool_definitions() -> List[ChatCompletionFunctionToolParam]: """Get OpenAI function definitions for memory tools. Returns: |