Python SDK

注意

Python SDK 正在开发中，后续版本可能会更改API。欢迎贡献代码：https://github.com/gorse-io/PyGorse

安装

• 选项 1：从 PyPI 安装：

pip install PyGorse

• 选项 2：从源代码安装：

git clone https://github.com/gorse-io/PyGorse.git
cd PyGorse
pip install .

用法

默认客户端Gorse是阻塞客户端。

from gorse import Gorse

# Create a blocking client.
client = Gorse('http://127.0.0.1:8087', 'api_key')

# Insert a user.
client.insert_user({
    'UserId': 'bob',
    'Labels': {
        'company': 'gorse',
        'location': 'hangzhou, china'
    },
    'Comment': 'Bob is a software engineer.'
})

# Insert an item.
client.insert_item({
    'ItemId': 'gorse-io:gorse',
    'IsHidden': False,
    'Labels': {
        'topics': ['recommendation', 'machine-learning']
    },
    'Categories': ['go'],
    'Timestamp': '2022-02-22',
    'Comment': 'Gorse is an open-source recommender system.'
})

# Insert feedbacks.
client.insert_feedbacks([
    { 'FeedbackType': 'star', 'UserId': 'bob', 'ItemId': 'ollama:ollama', 'Value': 1.0, 'Timestamp': '2022-02-24' },
    { 'FeedbackType': 'star', 'UserId': 'bob', 'ItemId': 'huggingface:transformers', 'Value': 1.0, 'Timestamp': '2022-02-25' },
    { 'FeedbackType': 'star', 'UserId': 'bob', 'ItemId': 'rasbt:llms-from-scratch', 'Value': 1.0, 'Timestamp': '2022-02-26' },
    { 'FeedbackType': 'star', 'UserId': 'bob', 'ItemId': 'vllm-project:vllm', 'Value': 1.0, 'Timestamp': '2022-02-27' },
    { 'FeedbackType': 'star', 'UserId': 'bob', 'ItemId': 'hiyouga:llama-factory', 'Value': 1.0, 'Timestamp': '2022-02-28' }
])

# Get recommendation.
client.get_recommend('bob', n=10)

AsyncGorse是一个异步客户端。

import asyncio

from gorse import AsyncGorse

async def main():
    # Create an async client.
    client = AsyncGorse('http://127.0.0.1:8087', 'api_key')

    # Get recommendation.
    await client.get_recommend('bob', n=10)

asyncio.run(main())