pygeai 0.1.6__py3-none-any.whl → 0.6.0b15__py3-none-any.whl

pygeai/tests/snippets/assistants/create_chat_assistant.py

@@ -0,0 +1,54 @@
+from pygeai.assistant.managers import AssistantManager
+from pygeai.core.models import LlmSettings, WelcomeData, WelcomeDataFeature, \
+    WelcomeDataExamplePrompt, GuardrailSettings, ChatAssistant
+
+from pygeai.core.services.llm.model import Model
+from pygeai.core.services.llm.providers import Provider
+
+
+manager = AssistantManager()
+
+guardrail_settings = GuardrailSettings(
+    llm_output=True,
+    input_moderation=True,
+    prompt_injection=True
+)
+
+llm_settings = LlmSettings(
+    provider_name=Provider.OPENAI,
+    model_name=Model.OpenAI.GPT_4,
+    temperature=0.7,
+    max_tokens=1000,
+    upload_files=False,
+    guardrail_settings=guardrail_settings
+)
+
+welcome_data = WelcomeData(
+    title="Assistant with WelcomeData",
+    description="It is to test WelcomeData",
+    features=[
+        WelcomeDataFeature(title="First Feature", description="First Feature Description"),
+        WelcomeDataFeature(title="Second Feature", description="Second Feature Description")
+    ],
+    examples_prompt=[
+        WelcomeDataExamplePrompt(
+            title="First Prompt Example",
+            description="First Prompt Example Description",
+            prompt_text="You are an assistant specialized in translating"
+        )
+    ]
+)
+
+assistant = ChatAssistant(
+    name="Welcome data Assistant 5",
+    description="A chatbot assistant for customer support",
+    prompt="Translate to French",
+    llm_settings=llm_settings,
+    welcome_data=welcome_data
+)
+
+try:
+    response = manager.create_assistant(assistant)
+    print(f"Response: {response}")
+except Exception as e:
+    print(f"Error: {e}")

pygeai/tests/snippets/assistants/create_text_assistant.py

@@ -0,0 +1,51 @@
+from pygeai.assistant.managers import AssistantManager
+from pygeai.core.models import TextAssistant, LlmSettings, WelcomeData, WelcomeDataFeature, \
+    WelcomeDataExamplePrompt, GuardrailSettings
+from pygeai.core.services.llm.model import Model
+from pygeai.core.services.llm.providers import Provider
+
+
+manager = AssistantManager()
+
+guardrail_settings = GuardrailSettings(
+    llm_output=True,
+    input_moderation=True,
+    prompt_injection=True
+)
+
+llm_settings = LlmSettings(
+    provider_name=Provider.OPENAI,
+    model_name=Model.OpenAI.GPT_4,
+    temperature=0.7,
+    max_tokens=1000,
+    upload_files=False,
+    guardrail_settings=guardrail_settings
+)
+
+welcome_data = WelcomeData(
+    title="AI Assistant",
+    description="An AI-powered assistant to help with various tasks.",
+    features=[
+        WelcomeDataFeature(title="Feature 1", description="Description of feature 1"),
+        WelcomeDataFeature(title="Feature 2", description="Description of feature 2")
+    ],
+    examples_prompt=[
+        WelcomeDataExamplePrompt(title="Example 1", description="This is an example", prompt_text="How can I help you today?"),
+        WelcomeDataExamplePrompt(title="Example 2", description="Another example", prompt_text="Tell me a joke.")
+    ]
+)
+
+assistant = TextAssistant(
+    name="ChatBot",
+    description="A chatbot assistant for customer support",
+    prompt="Hello! How can I assist you today?",
+    llm_settings=llm_settings,
+    welcome_data=welcome_data
+)
+
+
+try:
+    response = manager.create_assistant(assistant)
+    print(f"Response: {response}")
+except Exception as e:
+    print(f"Error: {e}")

pygeai/tests/snippets/assistants/data_analyst/extend_and_check.py

@@ -0,0 +1,100 @@
+import time
+from pathlib import Path
+from typing import List
+
+from pygeai.assistant.data_analyst.clients import DataAnalystAssistantClient
+from pygeai.core.common.exceptions import APIError
+
+
+def monitor_dataset_extension(
+    client: DataAnalystAssistantClient,
+    assistant_id: str,
+    file_paths: List[str],
+    poll_interval: float = 5.0,
+    max_attempts: int = 60
+) -> dict:
+    """
+    Uploads dataset files to a Data Analyst Assistant and monitors the processing status.
+
+    :param client: DataAnalystAssistantClient - The client instance to interact with the API.
+    :param assistant_id: str - The ID of the Data Analyst Assistant.
+    :param file_paths: List[str] - List of paths to .csv files to upload.
+    :param poll_interval: float - Time (in seconds) to wait between status checks (default: 5.0).
+    :param max_attempts: int - Maximum number of status checks before timing out (default: 60).
+    :return: dict - Final status response from the API.
+    :raises ValueError: If assistant_id or file_paths are invalid.
+    :raises FileNotFoundError: If any file in file_paths does not exist.
+    :raises APIError: If the API returns an error during upload or status checking.
+    """
+    # Validate inputs
+    if not assistant_id or not isinstance(assistant_id, str):
+        raise ValueError("assistant_id must be a non-empty string")
+    if not file_paths or not isinstance(file_paths, list):
+        raise ValueError("file_paths must be a non-empty list of file paths")
+
+    # Validate file paths
+    for file_path in file_paths:
+        path = Path(file_path)
+        if not path.is_file():
+            raise FileNotFoundError(f"File not found: {file_path}")
+        if path.suffix.lower() != '.csv':
+            raise ValueError(f"File must be a .csv file: {file_path}")
+
+    # Step 1: Upload the dataset files
+    print(f"Uploading files: {file_paths}")
+    try:
+        upload_response = client.extend_dataset(
+            assistant_id=assistant_id,
+            file_paths=file_paths
+        )
+        if isinstance(upload_response, dict) and "error" in upload_response:
+            raise APIError(f"Failed to upload files: {upload_response.get('error')}")
+        print(f"Upload successful: {upload_response}")
+    except Exception as e:
+        raise APIError(f"Upload failed: {str(e)}")
+
+    # Step 2: Poll the status until no longer 'IN PROGRESS'
+    print(f"Monitoring status for assistant ID: {assistant_id}")
+    attempts = 0
+    while attempts < max_attempts or max_attempts == 0:
+        try:
+            status_response = client.get_status(assistant_id=assistant_id)
+            if not isinstance(status_response, dict):
+                raise APIError(f"Invalid status response: {status_response}")
+
+            status = status_response.get("status")
+            percentage = status_response.get("percentage", "N/A")
+            print(f"Attempt {attempts + 1}/{max_attempts}: Status = {status}, Percentage = {percentage}")
+
+            if status != "IN PROGRESS":
+                return status_response
+
+            time.sleep(poll_interval)
+            attempts += 1
+        except Exception as e:
+            print(f"Error checking status: {str(e)}")
+            time.sleep(poll_interval)
+            attempts += 1
+
+    # Timeout if max_attempts reached
+    raise TimeoutError(f"Status check timed out after {max_attempts} attempts")
+
+
+if __name__ == "__main__":
+    try:
+        client = DataAnalystAssistantClient()
+        assistant_id = "bab64d47-6416-4f26-a0f4-765db7416652"
+        file_paths = [
+            "./magic_burgers_sales_data.csv",
+        ]
+
+        final_status = monitor_dataset_extension(
+            client=client,
+            assistant_id=assistant_id,
+            file_paths=file_paths,
+            poll_interval=5.0,
+            max_attempts=60
+        )
+        print(f"Final status: {final_status}")
+    except Exception as e:
+        print(f"Error: {str(e)}")

pygeai/tests/snippets/assistants/data_analyst/extend_dataset.py

@@ -0,0 +1,9 @@
+from pygeai.assistant.data_analyst.clients import DataAnalystAssistantClient
+
+client = DataAnalystAssistantClient()
+
+result = client.extend_dataset(
+    assistant_id="bab64d47-6416-4f26-a0f4-765db7416652",
+    file_paths=["./magic_burgers_sales_data.csv"]
+)
+print(result)

pygeai/tests/snippets/assistants/data_analyst/get_status.py

@@ -0,0 +1,9 @@
+from pygeai.assistant.data_analyst.clients import DataAnalystAssistantClient
+
+client = DataAnalystAssistantClient()
+try:
+    status = client.get_status(assistant_id="bab64d47-6416-4f26-a0f4-765db7416652")
+    print(status)
+except ValueError as e:
+    print(f"Error: {e}")
+

pygeai/tests/snippets/assistants/file_summarizer_assistant.py

@@ -0,0 +1,149 @@
+from uuid import uuid4
+from pygeai.chat.managers import ChatManager
+from pygeai.core.files.responses import UploadFileResponse
+from pygeai.core.files.managers import FileManager
+from pygeai.core.files.models import UploadFile
+from pygeai.core.models import ChatMessageList, ChatMessage, LlmSettings
+from pygeai.assistant.managers import AssistantManager
+from pygeai.core.services.llm.model import Model
+from pygeai.core.services.llm.providers import Provider
+from pygeai.core.models import WelcomeData, WelcomeDataFeature, WelcomeDataExamplePrompt, GuardrailSettings, ChatAssistant
+
+project_id = "2ca6883f-6778-40bb-bcc1-85451fb11107"
+
+file_manager = FileManager(
+    organization_id="4aa15b61-d3c7-4a5c-99b8-052d18a04ff2",
+    project_id=project_id
+)
+chat_manager = ChatManager()
+assistant_manager = AssistantManager()
+
+created_entities = {
+    "assistant_id": None,
+    "file_id": None
+}
+
+
+def rollback():
+    print("\n=== Initiating Rollback ===")
+    if created_entities["assistant_id"]:
+        print(f"Deleting assistant {created_entities['assistant_id']}...")
+        result = assistant_manager.delete_assistant(assistant_id=created_entities["assistant_id"])
+        print(f"Rollback: {result}")
+    if created_entities["file_id"]:
+        print(f"Deleting file {created_entities['file_id']}...")
+        result = file_manager.delete_file(file_id=created_entities["file_id"])
+        print(f"Rollback: {result}")
+    print("Rollback complete.")
+
+
+def main():
+    # File Upload Flow
+    print("\n=== File Upload Flow ===")
+    print("Uploading file 'test.txt'...")
+    file_to_upload = UploadFile(
+        name="test.txt",
+        path="./test.txt",
+        folder="summaries"
+    )
+    upload_result = file_manager.upload_file(file=file_to_upload)
+    if isinstance(upload_result, UploadFileResponse) and upload_result.success:
+        print(f"Success: Uploaded File: test.txt, ID: {upload_result.id}")
+        created_entities["file_id"] = upload_result.id
+    else:
+        print("Error: File upload failed:", upload_result)
+        rollback()
+        exit()
+
+    # Retrieve File Metadata
+    print("Retrieving file metadata...")
+    file_data = file_manager.get_file_data(file_id=created_entities["file_id"])
+    if hasattr(file_data, "name"):
+        file_name = file_data.name
+        print(f"Success: Retrieved File Name: {file_name}")
+    else:
+        print("Error: File retrieval failed:", file_data)
+        rollback()
+        exit()
+
+    # Assistant Flow
+    print("\n=== Assistant Flow ===")
+    print("Creating assistant 'FileSummaryAssistant'...")
+    assistant_id = str(uuid4())
+    guardrail_settings = GuardrailSettings(
+        llm_output=True,
+        input_moderation=True,
+        prompt_injection=True
+    )
+    llm_settings = LlmSettings(
+        provider_name=Provider.OPENAI,
+        model_name=Model.OpenAI.GPT_4,
+        temperature=0.7,
+        max_tokens=1000,
+        upload_files=True,
+        guardrail_settings=guardrail_settings
+    )
+    welcome_data = WelcomeData(
+        title="File Summary Assistant",
+        description="An assistant for summarizing files",
+        features=[
+            WelcomeDataFeature(title="File Summarization", description="Summarizes content from uploaded text files"),
+            WelcomeDataFeature(title="Quick Processing", description="Generates concise summaries in under 50 words")
+        ],
+        examples_prompt=[
+            WelcomeDataExamplePrompt(
+                title="Summarize File",
+                description="Summarize the content of an uploaded file",
+                prompt_text=f"Summarize the content of the file {{file:{file_name}}} in 50 words or less."
+            )
+        ]
+    )
+    assistant = ChatAssistant(
+        id=assistant_id,
+        name="FileSummaryAssistant2",
+        description="An assistant for summarizing text file content",
+        prompt="Read the provided text file by its ID and generate a concise summary in 50 words or less.",
+        llm_settings=llm_settings,
+        welcome_data=welcome_data
+    )
+    create_assistant_result = assistant_manager.create_assistant(assistant)
+    create_assistant_result = create_assistant_result.assistant if hasattr(create_assistant_result, "assistant") else create_assistant_result
+    if isinstance(create_assistant_result, ChatAssistant):
+        print(f"Success: Created Assistant: {create_assistant_result.name}, ID: {create_assistant_result.id}")
+        created_entities["assistant_id"] = create_assistant_result.id
+    else:
+        print("Error: Assistant creation failed:", create_assistant_result)
+        rollback()
+        exit()
+
+    # Chat Completion Flow
+    print("\n=== Chat Completion Flow ===")
+    print(f"Asking assistant about the uploaded file '{file_name}'...")
+    messages = ChatMessageList(messages=[
+        ChatMessage(
+            role="user",
+            content=f"Please summarize the content of the {{file:{file_name}}} ({file_name}) in 50 words or less."
+        )
+    ])
+    chat_response = chat_manager.chat_completion(
+        model=f"saia:assistant:{assistant.name}",
+        messages=messages,
+        llm_settings=llm_settings
+    )
+    if hasattr(chat_response, "choices") and chat_response.choices:
+        print(f"Success: Chat Response: {chat_response.choices[0].message.content}")
+    else:
+        print("Error: Chat completion failed:", chat_response)
+        rollback()
+        exit()
+
+    print("\n=== Process Completed Successfully ===")
+
+
+if __name__ == "__main__":
+    try:
+        main()
+        rollback()
+    except Exception as e:
+        rollback()
+        print(f"\n# Critical error: {e}")

pygeai/tests/snippets/assistants/get_assistant_data.py

@@ -0,0 +1,8 @@
+from pygeai.assistant.managers import AssistantManager
+
+
+manager = AssistantManager()
+
+
+response = manager.get_assistant_data(assistant_id="3372b866-36b3-4541-8897-d371dab99f10")
+print(f"response: {response}")

pygeai/tests/snippets/assistants/get_assistant_list.py

@@ -0,0 +1,7 @@
+from pygeai.organization.managers import OrganizationManager
+
+manager = OrganizationManager()
+
+
+response = manager.get_assistant_list("full")
+print(f"response: {response}")

pygeai/tests/snippets/assistants/rag/create_rag_assistant.py

@@ -0,0 +1,65 @@
+from pygeai.assistant.managers import AssistantManager
+from pygeai.core.models import WelcomeData, LlmSettings
+from pygeai.assistant.rag.models import Search, RetrieverOptions, SearchOptions, ChunkOptions, IndexOptions, \
+    RAGAssistant
+
+manager = AssistantManager()
+
+llm_options = LlmSettings(
+    cache=False,
+    temperature=0.1,
+    max_tokens=999,
+    model_name="gpt-3.5-turbo-16k",
+    n=1,
+    presence_penalty=0,
+    frequency_penalty=0,
+    provider="OpenAI",
+    stream=False,
+    top_p=1.0,
+    type=None,
+    verbose=True
+)
+
+retriever_options = RetrieverOptions(
+    type="vectorStore"
+)
+
+search_options = SearchOptions(
+    history_count=2,
+    llm=llm_options,
+    search=Search(
+        k=5,
+        return_source_documents=False,
+        score_threshold=0,
+        prompt="Use {context} and {question}",
+        template=""
+    ),
+    retriever=retriever_options
+)
+
+chunk_options = ChunkOptions(
+    chunk_size=999,
+    chunk_overlap=0
+)
+
+index_options = IndexOptions(
+    chunks=chunk_options
+)
+
+welcome_data = WelcomeData(
+    title="Test Profile Welcome Data",
+    description="Test Profile with WelcomeData",
+    features=[],
+    examples_prompt=[]
+)
+
+rag_assistant = RAGAssistant(
+    name="TestRAG3",
+    description="Test Profile with WelcomeData",
+    search_options=search_options,
+    index_options=index_options,
+    welcome_data=welcome_data
+)
+
+response = manager.create_assistant(rag_assistant)
+print(f"response: {response}")

pygeai/tests/snippets/assistants/rag/delete_al_documents.py

@@ -0,0 +1,7 @@
+from pygeai.assistant.managers import AssistantManager
+
+
+manager = AssistantManager()
+
+response = manager.delete_all_documents(name="Test-Profile-WelcomeData-3")
+print(response)

pygeai/tests/snippets/assistants/rag/delete_document.py

@@ -0,0 +1,10 @@
+from pygeai.assistant.managers import AssistantManager
+
+
+manager = AssistantManager()
+
+response = manager.delete_document(
+    name="Test-Profile-WelcomeData-4",
+    document_id="ce6f779e-329d-4e39-b84f-07a77265ff02"
+)
+print(response)

pygeai/tests/snippets/assistants/rag/delete_rag_assistant.py

@@ -0,0 +1,8 @@
+from pygeai.assistant.managers import AssistantManager
+from pygeai.organization.managers import OrganizationManager
+
+manager = AssistantManager()
+
+
+response = manager.delete_assistant(assistant_name="Test-Profile-WelcomeData-9")
+print(f"response: {response}")

pygeai/tests/snippets/assistants/rag/get_document.py

@@ -0,0 +1,10 @@
+from pygeai.assistant.managers import AssistantManager
+
+
+manager = AssistantManager()
+
+response = manager.get_document(
+    name="Test-Profile-WelcomeData-4",
+    document_id="ce6f779e-329d-4e39-b84f-07a77265ff02"
+)
+print(response)

pygeai/tests/snippets/assistants/rag/get_documents.py

@@ -0,0 +1,7 @@
+from pygeai.assistant.managers import AssistantManager
+from pygeai.organization.managers import OrganizationManager
+
+manager = AssistantManager()
+
+response = manager.get_document_list(name="Test-Profile-WelcomeData-4")
+print(response)

pygeai/tests/snippets/assistants/rag/get_rag_assistant_data.py

@@ -0,0 +1,8 @@
+from pygeai.assistant.managers import AssistantManager
+from pygeai.organization.managers import OrganizationManager
+
+manager = AssistantManager()
+
+
+response = manager.get_assistant_data(assistant_name="Test-Profile-WelcomeData-2")
+print(f"response: {response}")

pygeai/tests/snippets/assistants/rag/update_rag_assistant.py

@@ -0,0 +1,48 @@
+from pygeai.assistant.managers import AssistantManager
+from pygeai.core.models import WelcomeDataFeature, WelcomeDataExamplePrompt, WelcomeData
+from pygeai.assistant.rag.models import RAGAssistant
+
+
+features = [
+    WelcomeDataFeature(
+        title="Updated weather conditions",
+        description="Get the state of the weather in any country"
+    ),
+    WelcomeDataFeature(
+        title="Rain probability updated",
+        description="Get the rain probability in any location"
+    )
+]
+
+examples_prompt = [
+    WelcomeDataExamplePrompt(
+        title="First Prompt Example",
+        description="First Prompt Example Description",
+        prompt_text="You are an assistant specializing in..."
+    )
+]
+
+welcome_data = WelcomeData(
+    title="Welcome to RAG Update",
+    description="It is a RAG created with WelcomeData via API",
+    features=features,
+    examples_prompt=examples_prompt
+)
+
+rag_assistant = RAGAssistant(
+    name="Test-Profile-WelcomeData-2",
+    status=1,
+    description="Updated RAGAssistant profile with welcome data",
+    template="",
+    welcome_data=welcome_data,
+    search_options=None,
+    index_options=None
+)
+
+manager = AssistantManager()
+
+updated_assistant = manager.update_assistant(assistant=rag_assistant)
+
+print(updated_assistant)
+
+

pygeai/tests/snippets/assistants/rag/upload_document.py

@@ -0,0 +1,19 @@
+from pygeai.assistant.managers import AssistantManager
+from pygeai.assistant.rag.models import UploadDocument, RAGAssistant, UploadType
+
+manager = AssistantManager()
+
+assistant = RAGAssistant(
+    name="Test-Profile-WelcomeData-4"
+)
+
+document = UploadDocument(
+    path="test.txt",
+    upload_type=UploadType.MULTIPART,
+    metadata={'type': 'test', 'year': 2025},
+    content_type="text/plain"
+)
+
+response = manager.upload_document(assistant=assistant, document=document)
+
+print(response)

pygeai/tests/snippets/assistants/send_feedback.py

@@ -0,0 +1,14 @@
+from pygeai.assistant.managers import AssistantManager
+from pygeai.core.feedback.models import FeedbackRequest
+
+feedback_request = FeedbackRequest(
+    request_id="28adc1a4-8a21-4d3b-94e2-201534408528",
+    origin="user-feedback",
+    answer_score=1,
+    comments="Great response!"
+)
+
+assistant_manager = AssistantManager()
+
+response = assistant_manager.send_feedback(feedback_request=feedback_request)
+print(response)

pygeai/tests/snippets/assistants/update_chat_assistant.py

@@ -0,0 +1,63 @@
+from pygeai.assistant.managers import AssistantManager
+from pygeai.core.models import LlmSettings, WelcomeData, WelcomeDataFeature, \
+    WelcomeDataExamplePrompt, GuardrailSettings, ChatAssistant
+from pygeai.core.services.llm.model import Model
+from pygeai.core.services.llm.providers import Provider
+
+
+client = AssistantManager()
+
+guardrail_settings = GuardrailSettings(
+    llm_output=True,
+    input_moderation=True,
+    prompt_injection=True
+)
+
+llm_settings = LlmSettings(
+    provider_name=Provider.OPENAI,
+    model_name=Model.OpenAI.GPT_4,
+    temperature=0.7,
+    max_tokens=1000,
+    upload_files=False,
+    guardrail_settings=guardrail_settings
+)
+
+welcome_data = WelcomeData(
+    title="Assistant with WelcomeData",
+    description="It is to test WelcomeData",
+    features=[
+        WelcomeDataFeature(title="First Feature", description="First Feature Description"),
+        WelcomeDataFeature(title="Second Feature", description="Second Feature Description")
+    ],
+    examples_prompt=[
+        WelcomeDataExamplePrompt(
+            title="First Prompt Example",
+            description="First Prompt Example Description",
+            prompt_text="You are an assistant specialized in translating"
+        ),
+        WelcomeDataExamplePrompt(
+            title="Second Prompt Example",
+            description="Second Prompt Example Description",
+            prompt_text="You are an assistant specialized in transcribing"
+        ),
+
+    ]
+)
+
+assistant = ChatAssistant(
+    id="3372b866-36b3-4541-8897-d371dab99f10",
+    name="Welcome data Assistant 4",
+    description="A chatbot assistant for customer support",
+    prompt="Translate to French",
+    llm_settings=llm_settings,
+    welcome_data=welcome_data
+)
+
+try:
+    response = client.update_assistant(
+        assistant=assistant,
+        action="saveNewRevision"
+    )
+    print(f"Response: {response}")
+except Exception as e:
+    print(f"Error: {e}")