diff options
| author | Dhravya Shah <[email protected]> | 2025-10-03 02:41:49 -0700 |
|---|---|---|
| committer | Dhravya Shah <[email protected]> | 2025-10-03 02:41:49 -0700 |
| commit | 4d6fd37c99fd6af46d2f1aedbeb750e0029b3b8b (patch) | |
| tree | 869d415d87b152bfb5e55601311a2c8f6e64b6fc /packages/openai-sdk-python | |
| parent | chore: fix docs again (diff) | |
| download | supermemory-4d6fd37c99fd6af46d2f1aedbeb750e0029b3b8b.tar.xz supermemory-4d6fd37c99fd6af46d2f1aedbeb750e0029b3b8b.zip | |
fix: model names
Diffstat (limited to 'packages/openai-sdk-python')
| -rw-r--r-- | packages/openai-sdk-python/README.md | 22 | ||||
| -rw-r--r-- | packages/openai-sdk-python/tests/test_tools.py | 2 |
2 files changed, 12 insertions, 12 deletions
diff --git a/packages/openai-sdk-python/README.md b/packages/openai-sdk-python/README.md index fdb9fb84..fc0832e1 100644 --- a/packages/openai-sdk-python/README.md +++ b/packages/openai-sdk-python/README.md @@ -30,29 +30,29 @@ from supermemory_openai import SupermemoryTools, execute_memory_tool_calls async def main(): # Initialize OpenAI client client = openai.AsyncOpenAI(api_key="your-openai-api-key") - + # Initialize Supermemory tools tools = SupermemoryTools( api_key="your-supermemory-api-key", config={"project_id": "my-project"} ) - + # Chat with memory tools response = await client.chat.completions.create( - model="gpt-4o", + model="gpt-5", messages=[ { "role": "system", "content": "You are a helpful assistant with access to user memories." }, { - "role": "user", + "role": "user", "content": "Remember that I prefer tea over coffee" } ], tools=tools.get_tool_definitions() ) - + # Handle tool calls if present if response.choices[0].message.tool_calls: tool_results = await execute_memory_tool_calls( @@ -61,7 +61,7 @@ async def main(): config={"project_id": "my-project"} ) print("Tool results:", tool_results) - + print(response.choices[0].message.content) asyncio.run(main()) @@ -91,7 +91,7 @@ result = await tools.search_memories( include_full_docs=True ) -# Add memory +# Add memory result = await tools.add_memory( memory="User prefers tea over coffee" ) @@ -107,7 +107,7 @@ result = await tools.fetch_memory( ```python from supermemory_openai import ( create_search_memories_tool, - create_add_memory_tool, + create_add_memory_tool, create_fetch_memory_tool ) @@ -128,7 +128,7 @@ if response.choices[0].message.tool_calls: tool_calls=response.choices[0].message.tool_calls, config={"project_id": "my-project"} ) - + # Add tool results to conversation messages.append(response.choices[0].message) messages.extend(tool_results) @@ -163,7 +163,7 @@ SupermemoryTools( try: response = await client.chat_completion( messages=[{"role": "user", "content": "Hello"}], - model="gpt-4o" + model="gpt-5" ) except Exception as e: print(f"Error: {e}") @@ -175,7 +175,7 @@ Set these environment variables for testing: - `SUPERMEMORY_API_KEY` - Your Supermemory API key - `OPENAI_API_KEY` - Your OpenAI API key -- `MODEL_NAME` - Model to use (default: "gpt-4o-mini") +- `MODEL_NAME` - Model to use (default: "gpt-5-nano") - `SUPERMEMORY_BASE_URL` - Custom Supermemory base URL (optional) ## Development diff --git a/packages/openai-sdk-python/tests/test_tools.py b/packages/openai-sdk-python/tests/test_tools.py index 273c06e7..6ecb9d8f 100644 --- a/packages/openai-sdk-python/tests/test_tools.py +++ b/packages/openai-sdk-python/tests/test_tools.py @@ -71,7 +71,7 @@ def test_base_url() -> str: @pytest.fixture def test_model_name() -> str: """Get test model name from environment.""" - return os.getenv("MODEL_NAME", "gpt-4o-mini") + return os.getenv("MODEL_NAME", "gpt-5-nano") class TestToolInitialization: |