add-assistant #2
|
@ -2,7 +2,7 @@ import os
|
|||
import logging
|
||||
import pandas as pd
|
||||
from openai import OpenAI
|
||||
|
||||
import textwrap
|
||||
|
||||
class OpenAIResourceManager:
|
||||
"""
|
||||
|
@ -105,15 +105,15 @@ class OpenAIResourceManager:
|
|||
self.delete_all_vector_stores()
|
||||
self.delete_all_files()
|
||||
|
||||
def truncate_string(self, s, max_length=50):
|
||||
def wrap_text(self, s, width=30):
|
||||
"""
|
||||
Truncate a string to a maximum length with ellipsis.
|
||||
Wrap text to a specific width.
|
||||
|
||||
:param s: The string to truncate.
|
||||
:param max_length: The maximum length of the string.
|
||||
:return: The truncated string.
|
||||
:param s: The string to wrap.
|
||||
:param width: The maximum width of each line.
|
||||
:return: The wrapped string.
|
||||
"""
|
||||
return (s[:max_length] + "...") if len(s) > max_length else s
|
||||
return "\n".join(textwrap.wrap(s, width))
|
||||
|
||||
def show_all_assistants(self):
|
||||
"""
|
||||
|
@ -122,7 +122,7 @@ class OpenAIResourceManager:
|
|||
assistants = self.get_all_assistants()
|
||||
assistant_data = [
|
||||
{
|
||||
k: self.truncate_string(str(v), max_length=25)
|
||||
k: self.wrap_text(str(v))
|
||||
for k, v in assistant.dict().items()
|
||||
}
|
||||
for assistant in assistants
|
||||
|
@ -137,7 +137,7 @@ class OpenAIResourceManager:
|
|||
"""
|
||||
vector_stores = self.get_all_vector_stores()
|
||||
vector_store_data = [
|
||||
{k: self.truncate_string(str(v)) for k, v in vector_store.dict().items()}
|
||||
{k: self.wrap_text(str(v)) for k, v in vector_store.dict().items()}
|
||||
for vector_store in vector_stores
|
||||
]
|
||||
df = pd.DataFrame(vector_store_data)
|
||||
|
@ -150,13 +150,22 @@ class OpenAIResourceManager:
|
|||
"""
|
||||
files = self.get_all_files()
|
||||
file_data = [
|
||||
{k: self.truncate_string(str(v)) for k, v in file.dict().items()}
|
||||
{k: self.wrap_text(str(v)) for k, v in file.dict().items()}
|
||||
for file in files
|
||||
]
|
||||
df = pd.DataFrame(file_data)
|
||||
print("Files:")
|
||||
print(df.to_markdown(index=False))
|
||||
|
||||
def show_all_resources(self):
|
||||
"""
|
||||
Display all resources in a table.
|
||||
"""
|
||||
self.show_all_assistants()
|
||||
print("")
|
||||
self.show_all_vector_stores()
|
||||
print("")
|
||||
self.show_all_files()
|
||||
|
||||
if __name__ == "__main__":
|
||||
# Get the OpenAI API key from the environment variables
|
||||
|
@ -164,7 +173,6 @@ if __name__ == "__main__":
|
|||
# Create an instance of the OpenAIResourceManager
|
||||
manager = OpenAIResourceManager(api_key=api_key)
|
||||
|
||||
# Show all resources in tables
|
||||
manager.show_all_assistants()
|
||||
manager.show_all_vector_stores()
|
||||
manager.show_all_files()
|
||||
# Show or delete all resources in tables
|
||||
manager.show_all_resources()
|
||||
# manager.delete_all_resources()
|
||||
|
|
Loading…
Reference in New Issue