This commit is contained in:
Sarah Wooders
2023-10-23 15:58:11 -07:00
parent 62cb3eda60
commit 9309c6bc2f
2 changed files with 5 additions and 2 deletions

View File

@@ -15,7 +15,7 @@ import sqlite3
import fitz
from tqdm import tqdm
from memgpt.openai_tools import async_get_embedding_with_backoff
from memgpt.configs import memgpt_dir
from memgpt.config import memgpt_dir
def count_tokens(s: str, model: str = "gpt-4") -> int:
encoding = tiktoken.encoding_for_model(model)

View File

@@ -1,6 +1,9 @@
[tool.poetry]
name = "memgpt"
name = "pymemgpt"
version = "0.1.0"
packages = [
{include = "memgpt"}
]
description = "Teaching LLMs memory management for unbounded context"
authors = [
"Charles Packer <contact@charlespacker.com>",