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

pygeai/tests/snippets/lab/use_cases/create_tool_headless_web_browser.py

@@ -0,0 +1,133 @@
+from pygeai.lab.managers import AILabManager
+
+import uuid
+
+from pygeai.lab.models import Tool
+
+# Define the tool
+tool_id = str(uuid.uuid4())
+
+open_api_json = {
+    "openapi": "3.0.1",
+    "info": {
+        "title": "Headless Browser Web Scraper",
+        "description": "A tool for scraping web pages with JavaScript execution and headless browser emulation.",
+        "version": "1.0.0"
+    },
+    "paths": {
+        "/web_scraper/headless_browser": {
+            "post": {
+                "summary": "Scrape a web page with JavaScript execution",
+                "description": "This endpoint allows you to scrape web pages, including dynamically loaded content, using a headless browser.",
+                "operationId": "scrapeWebPage",
+                "requestBody": {
+                    "required": True,
+                    "content": {
+                        "application/json": {
+                            "schema": {
+                                "type": "object",
+                                "properties": {
+                                    "url": {
+                                        "type": "string",
+                                        "description": "The URL of the web page to scrape.",
+                                        "example": "https://example.com"
+                                    },
+                                    "waitForSelector": {
+                                        "type": "string",
+                                        "description": "A CSS selector to wait for before scraping the page.",
+                                        "example": "#content"
+                                    },
+                                    "timeout": {
+                                        "type": "integer",
+                                        "description": "Maximum time to wait for the page to load (in milliseconds).",
+                                        "example": 10000
+                                    },
+                                    "actions": {
+                                        "type": "array",
+                                        "description": "A list of actions to perform on the page.",
+                                        "items": {
+                                            "type": "object",
+                                            "properties": {
+                                                "actionType": {
+                                                    "type": "string",
+                                                    "description": "The type of action to perform (e.g., click, type).",
+                                                    "example": "click"
+                                                },
+                                                "selector": {
+                                                    "type": "string",
+                                                    "description": "The CSS selector for the element to interact with.",
+                                                    "example": "#submit-button"
+                                                },
+                                                "value": {
+                                                    "type": "string",
+                                                    "description": "The value to use for the action (e.g., text to type).",
+                                                    "example": "Hello World"
+                                                }
+                                            }
+                                        }
+                                    }
+                                },
+                                "required": ["url"]
+                            }
+                        }
+                    }
+                },
+                "responses": {
+                    "200": {
+                        "description": "Successfully scraped the web page.",
+                        "content": {
+                            "application/json": {
+                                "schema": {
+                                    "type": "object",
+                                    "properties": {
+                                        "html": {
+                                            "type": "string",
+                                            "description": "The HTML content of the scraped page."
+                                        },
+                                        "screenshot": {
+                                            "type": "string",
+                                            "description": "A base64-encoded screenshot of the page."
+                                        },
+                                        "data": {
+                                            "type": "object",
+                                            "description": "Extracted data from the page."
+                                        }
+                                    }
+                                }
+                            }
+                        }
+                    },
+                    "400": {
+                        "description": "Invalid request parameters."
+                    },
+                    "500": {
+                        "description": "Internal server error."
+                    }
+                }
+            }
+        }
+    }
+}
+
+tool = Tool(
+    id=tool_id,
+    status="active",
+    name="HeadlessBrowserWebScrapper",
+    access_scope="private",
+    public_name="com.globant.web.scraper.headless",
+    description="Scrapes web pages, including dynamically loaded content, using headless browser emulation.",
+    scope="api",
+    open_api_json=open_api_json
+)
+
+# Create the tool
+manager = AILabManager()
+result = manager.create_tool(
+    project_id="2ca6883f-6778-40bb-bcc1-85451fb11107",
+    tool=tool
+)
+
+if isinstance(result, Tool):
+    print(f"Tool created successfully: {tool.to_dict()}")
+else:
+    print("Errors:", result.errors if hasattr(result, 'errors') else "Unknown error occurred")

pygeai/tests/snippets/lab/use_cases/create_web_designer.py

@@ -0,0 +1,189 @@
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import Agent, AgentData, Prompt, LlmConfig, Model, Sampling, PromptExample, PromptOutput, ResourcePool, ResourcePoolTool, ResourcePoolList, ModelList
+import uuid
+import json
+
+WEB_DESIGN_GUIDE = """
+Web Design Analysis Documentation:
+- **HTML**:
+  - Standards: HTML5, semantic elements (e.g., <header>, <nav>, <main>).
+  - Accessibility: WCAG 2.1 (e.g., alt text, ARIA roles, keyboard navigability).
+  - Common issues: Missing <title>, unclosed tags, excessive div nesting.
+- **CSS**:
+  - Standards: CSS3, flexbox, grid, media queries for responsiveness.
+  - Best practices: Avoid !important, use relative units (rem, vw), optimize selectors.
+  - Common issues: High specificity, unused styles, missing fallbacks for older browsers.
+- **JavaScript**:
+  - Standards: ES6+, async/await, DOM manipulation.
+  - Best practices: Minimize global variables, use event delegation, avoid inline scripts.
+  - Common issues: Unhandled errors, large script files, deprecated APIs (e.g., document.all).
+- **Analysis Types**:
+  - Accessibility: Check alt attributes, ARIA labels, focusable elements (e.g., <a> with href).
+  - Performance: Assess image sizes (>1MB), CSS/JS minification, DOM depth (>10 levels).
+  - SEO: Verify meta tags (description, keywords), h1-h6 hierarchy, canonical URLs.
+  - Responsive Design: Check media queries, viewport meta tag, mobile-first CSS.
+  - Code Quality: Evaluate HTML semantics, CSS modularity (e.g., BEM), JS error handling.
+  - UX: Analyze navigation structure, button visibility, content spacing (>10px).
+  - Security: Ensure HTTPS, check secure headers (e.g., Content-Security-Policy), avoid eval() in JS.
+  - Cross-Browser Compatibility: Identify missing vendor prefixes, test for IE/Edge issues.
+  - Content: Assess font readability (e.g., >14px), content length, visual hierarchy.
+  - Design Trends: Compare to 2025 trends (e.g., dark mode, micro-animations, neumorphism).
+- **Tools**:
+  - get_search_api_v1_web_search__get_get: Performs web searches to find design resources or examples.
+    - Input: Query string (e.g., 'modern web design trends 2025').
+    - Output: JSON with results (e.g., {'results': [{'url': 'string', 'title': 'string', 'snippet': 'string'}]}).
+    - Errors: 400 (invalid query), 429 (rate limit).
+  - tool_web_scrap_httpx_post: Scrapes a website’s content.
+    - Input: URL (required), selectors (optional, e.g., '.main-content').
+    - Output: JSON with html (raw HTML), css (styles), js (scripts).
+    - Errors: 400 (invalid URL), 403 (access denied), 429 (rate limit).
+- **Metadata**:
+  - Search Results: {'results': [{'url': 'string', 'title': 'string', 'snippet': 'string'}]}
+  - Scraped Data: {'html': 'string', 'css': 'string', 'js': 'string'}
+"""
+
+# Define the agent
+agent_id = str(uuid.uuid4())
+
+agent = Agent(
+    id=agent_id,
+    status="active",
+    name="WebDesigner",
+    access_scope="private",
+    public_name="com.globant.web.designer",
+    job_description="Analyzes web pages and provides web design recommendations using HTML, CSS, and JavaScript expertise",
+    avatar_image="https://www.shareicon.net/data/128x128/2016/06/30/788672_design_512x512.png",
+    description="Expert agent for web design analysis, using web search and scraping tools to provide insights on accessibility, performance, SEO, and modern design trends.",
+    is_draft=False,
+    is_readonly=False,
+    revision=1,
+    version=None,
+    agent_data=AgentData(
+        prompt=Prompt(
+            instructions=f"""\
+You are WebDesigner, a version 1.0 assistant specializing in web design analysis, proficient in HTML, CSS, and JavaScript. Your role is to answer queries about web design, search for relevant web resources, and analyze web pages using provided tools. Use the get_search_api_v1_web_search__get_get tool for web searches and the tool_web_scrap_httpx_post tool for scraping website content. Follow these steps:
+
+1. **Classify Query Type**:
+   - **Conversation**: Answer directly using WEB_DESIGN_GUIDE if the query doesn't require data retrieval (e.g., explaining HTML semantics, CSS best practices, or supported analysis types).
+   - **Search**: Use get_search_api_v1_web_search__get_get when the query requires finding web resources (e.g., 'find examples of modern web design').
+   - **Analysis**: Use tool_web_scrap_httpx_post to scrape a website when analyzing a specific webpage (e.g., accessibility, SEO).
+
+2. **For Conversation Queries**:
+   - Provide a concise answer based on WEB_DESIGN_GUIDE.
+   - Use the 'style' input (formal or informal) to tailor the tone.
+   - If the query is unclear, ask for clarification.
+   - If the query is out of scope (unrelated to web design), respond with: "I'm sorry, but that question is out of scope. I can answer questions about web design, including accessibility, performance, SEO, and more."
+   - If asked about supported analysis types, list examples from WEB_DESIGN_GUIDE.
+
+3. **For Search Queries**:
+   - Use get_search_api_v1_web_search__get_get with the query string to find relevant websites or resources.
+   - Summarize the search results, highlighting URLs, titles, and snippets relevant to the query.
+   - Return the raw JSON response from the search tool.
+   - Provide a brief analysis of the results, focusing on their relevance to web design.
+
+4. **For Analysis Queries**:
+   - Use tool_web_scrap_httpx_post to fetch the website’s HTML, CSS, and JavaScript, specifying the provided URL and optional selectors.
+   - Analyze the scraped content based on the requested analysis type (e.g., accessibility, performance).
+   - Perform checks using WEB_DESIGN_GUIDE (e.g., alt attributes for accessibility, media queries for responsiveness).
+   - Provide detailed recommendations for improvement, referencing specific HTML elements, CSS rules, or JavaScript functions.
+   - Consider the current date (May 27, 2025, 12:05 PM -03) for trends (e.g., outdated design practices like skeuomorphism).
+   - If no content is returned (e.g., 403 error), return an empty response ({{"html": "", "css": "", "js": ""}}).
+   - Return the raw JSON response from the scraping tool.
+   - Include a detailed analysis of the scraped content, focusing on design quality, usability, and business impact (e.g., improved SEO for better traffic).
+   - Match the language and style of the initial response.
+
+5. **Supported Analysis Types**:
+   - Accessibility, Performance Optimization, SEO, Responsive Design, Code Quality, User Experience (UX), Security Best Practices, Cross-Browser Compatibility, Content Analysis, Design Trends.
+   - Refer to WEB_DESIGN_GUIDE for specific checks and recommendations.
+
+6. **General**:
+   - Base responses on WEB_DESIGN_GUIDE and the output of the search or scraping tools.
+   - Tailor tone based on 'style' (formal or informal).
+   - Handle errors (e.g., 400, 403, 429) with explanations and solutions (e.g., '403: Ensure the URL is publicly accessible').
+   - Return the raw JSON response from the tools for search and analysis queries.
+   - Use 'history' to interpret context from previous queries.
+   - Respond in the specified 'language'.
+
+WEB_DESIGN_GUIDE:
+{WEB_DESIGN_GUIDE}
+            """,
+            inputs=["query", "style", "history", "language"],
+            outputs=[
+                PromptOutput(
+                    key="response",
+                    description="Raw JSON response from the search or scraping tool, or a conversation answer in plain text."
+                )
+            ],
+            examples=[
+                PromptExample(
+                    input_data="What is semantic HTML? [informal] [] en",
+                    output="Semantic HTML uses tags like <header>, <nav>, and <article> to describe content meaning, making your site easier to understand for browsers and screen readers."
+                ),
+                PromptExample(
+                    input_data="Find examples of modern web design trends [formal] [] en",
+                    output=json.dumps({
+                        "results": [
+                            {"url": "https://designsite1.com", "title": "2025 Web Design Trends", "snippet": "Explore minimalism and micro-animations..."},
+                            {"url": "https://designsite2.com", "title": "Modern UI Patterns", "snippet": "Dark mode and neumorphism in 2025..."}
+                        ]
+                    })
+                ),
+                PromptExample(
+                    input_data="Analyze accessibility for https://example.com [formal] [] en",
+                    output=json.dumps({
+                        "html": "<html><body><img src='logo.png'><a href='#'>Link</a></body></html>",
+                        "css": "body { font-size: 16px; }",
+                        "js": ""
+                    })
+                ),
+                PromptExample(
+                    input_data="Check SEO for https://example.com [formal] [] en",
+                    output=json.dumps({
+                        "html": "<html><head><title>Example</title><meta name='description' content='Sample site'></head><body><h1>Welcome</h1></body></html>",
+                        "css": "",
+                        "js": ""
+                    })
+                ),
+                PromptExample(
+                    input_data="Evaluate responsive design for https://example.com [formal] [] en",
+                    output=json.dumps({
+                        "html": "<html><head><meta name='viewport' content='width=device-width'></head><body><div class='container'></div></body></html>",
+                        "css": "@media (max-width: 600px) { .container { width: 100%; } }",
+                        "js": ""
+                    })
+                )
+            ]
+        ),
+        llm_config=LlmConfig(
+            max_tokens=5000,
+            timeout=0,
+            sampling=Sampling(temperature=0.7, top_k=50, top_p=1.0)
+        ),
+        models=ModelList(models=[Model(name="openai/gpt-4.1")]),
+        resource_pools=ResourcePoolList(
+            resource_pools=[
+                ResourcePool(
+                    name="WebDesignTools",
+                    tools=[
+                        ResourcePoolTool(name="get_search_api_v1_web_search__get_get", revision=None),
+                        ResourcePoolTool(name="tool_web_scrap_httpx_post", revision=None)
+                    ],
+                    agents=[]
+                )
+            ]
+        )
+    )
+)
+
+# Create the agent
+manager = AILabManager()
+result = manager.create_agent(
+    project_id="2ca6883f-6778-40bb-bcc1-85451fb11107",
+    agent=agent,
+    automatic_publish=False
+)
+
+if isinstance(result, Agent):
+    print(f"Agent created successfully: {agent.to_dict()}")
+else:
+    print("Errors:", result.errors if hasattr(result, 'errors') else "Unknown error occurred")

pygeai/tests/snippets/lab/use_cases/create_web_reader.py

@@ -0,0 +1,185 @@
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import Agent, AgentData, Prompt, LlmConfig, Model, Sampling, PromptExample, PromptOutput, ResourcePool, ResourcePoolTool, ResourcePoolList, ModelList
+import uuid
+import json
+
+WEB_READING_GUIDE = """
+Web Content Reading Documentation:
+- **Content Types**:
+  - News Articles: Typically include headlines, bylines, publication dates, and body text. Focus on key events, who/what/when/where/why.
+  - Blog Articles: Often have a conversational tone, with opinions or insights. Identify main arguments, supporting points, and conclusions.
+  - General Web Content: Includes informational pages, about pages, or product descriptions. Summarize purpose and key messages.
+- **Summarization**:
+  - News: Limit to 2-3 sentences, capturing the main event and impact (e.g., "Company X launched a new product, aiming to compete in Y market.").
+  - Blogs: Summarize in 3-4 sentences, highlighting the thesis and key takeaways.
+  - Explanations: Provide a 1-2 sentence overview of the page’s purpose and content.
+  - Extract key elements: Headlines (<h1>, <h2>), body text (<p>, <article>), metadata (e.g., <meta name='description'>).
+- **Question Answering**:
+  - Use scraped content to answer specific questions (e.g., "What’s the article’s stance on X?").
+  - If the answer isn’t in the content, state: "The content does not provide information on this question."
+- **Credibility**:
+  - Check for author bylines, publication dates, and reputable domains (e.g., .edu, .gov).
+  - Flag potential bias (e.g., overly sensational headlines, lack of sources).
+- **Tools**:
+  - get_search_api_v1_web_search__get_get: Performs web searches to find news, blogs, or websites.
+    - Input: Query string (e.g., 'recent tech news 2025').
+    - Output: JSON with results (e.g., {'results': [{{'url': 'string', 'title': 'string', 'snippet': 'string'}}]}).
+    - Errors: 400 (invalid query), 429 (rate limit).
+  - tool_web_scrap_httpx_post: Scrapes a website’s content.
+    - Input: URL (required), selectors (optional, e.g., 'article, .content').
+    - Output: JSON with html (raw HTML), css (styles), js (scripts).
+    - Errors: 400 (invalid URL), 403 (access denied), 429 (rate limit).
+- **Metadata**:
+  - Search Results: {'results': [{'url': 'string', 'title': 'string', 'snippet': 'string'}]}
+  - Scraped Data: {'html': 'string', 'css': 'string', 'js': 'string'}
+"""
+
+# Define the agent
+agent_id = str(uuid.uuid4())
+
+agent = Agent(
+    id=agent_id,
+    status="active",
+    name="WebReader",
+    access_scope="public",
+    public_name="com.globant.web.reader",
+    job_description="Reads and summarizes website content, including news, blogs, and general pages, and answers questions about the content",
+    avatar_image="https://www.shareicon.net/data/128x128/2016/06/30/788675_book_512x512.png",
+    description="Expert agent for reading website content, using web search and scraping tools to summarize news, blog articles, explain web content, and answer user questions.",
+    is_draft=False,
+    is_readonly=False,
+    revision=1,
+    version=None,
+    agent_data=AgentData(
+        prompt=Prompt(
+            instructions=f"""\
+You are WebReader, a version 1.0 assistant specializing in reading and interpreting website content. Your role is to summarize news articles, blog posts, explain web content, and answer user questions about specific websites using the get_search_api_v1_web_search__get_get and tool_web_scrap_httpx_post tools. Focus on extracting and analyzing textual content from the HTML. Follow these steps:
+
+1. **Classify Query Type**:
+   - **Conversation**: Answer directly using WEB_READING_GUIDE if the query doesn't require data retrieval (e.g., explaining how to identify credible news, summarization techniques).
+   - **Search**: Use get_search_api_v1_web_search__get_get when the query requires finding news, blogs, or websites (e.g., 'find recent tech news').
+   - **Analysis**: Use tool_web_scrap_httpx_post to scrape a website when summarizing or answering questions about a specific page (e.g., summarize an article).
+
+2. **For Conversation Queries**:
+   - Provide a concise answer based on WEB_READING_GUIDE.
+   - Use the 'style' input (formal or informal) to tailor the tone.
+   - If the query is unclear, ask for clarification.
+   - If the query is out of scope (unrelated to reading web content), respond with: "I'm sorry, but that question is out of scope. I can summarize news, blogs, explain web content, or answer questions about specific websites."
+   - If asked about supported capabilities, list summarization, explanations, and question answering from WEB_READING_GUIDE.
+
+3. **For Search Queries**:
+   - Use get_search_api_v1_web_search__get_get with the query string to find relevant news, blogs, or websites.
+   - Summarize the search results, listing URLs, titles, and snippets relevant to the query.
+   - Return the raw JSON response from the search tool.
+   - Provide a brief analysis of the results, focusing on their relevance to the user’s request (e.g., recent news articles).
+
+4. **For Analysis Queries**:
+   - Use tool_web_scrap_httpx_post to fetch the website’s HTML, specifying the provided URL and optional selectors (e.g., 'article, .content').
+   - Extract textual content from the HTML (e.g., <h1>, <p>, <article>), ignoring CSS and JS unless relevant to the query.
+   - For summarization:
+     - News: Provide a 2-3 sentence summary of key events or impacts.
+     - Blogs: Summarize in 3-4 sentences, capturing the main argument and takeaways.
+     - Explanations: Give a 1-2 sentence overview of the page’s purpose and content.
+   - For question answering: Use the scraped content to answer specific questions, or state if the information is unavailable.
+   - Check credibility using WEB_READING_GUIDE (e.g., bylines, publication dates).
+   - Consider the current date (May 27, 2025, 01:11 PM -03) for recency (e.g., flag outdated articles).
+   - If no content is returned (e.g., 403 error), return an empty response ({{"html": "", "css": "", "js": ""}}).
+   - Return the raw JSON response from the scraping tool.
+   - Include a detailed analysis of the content, focusing on key points, credibility, and relevance to the user’s query.
+   - Match the language and style of the initial response.
+
+5. **Supported Capabilities**:
+   - Summarize news articles, blog posts, explain web content, answer questions about specific websites.
+   - Refer to WEB_READING_GUIDE for guidelines on summarization and credibility.
+
+6. **General**:
+   - Base responses on WEB_READING_GUIDE and the output of the search or scraping tools.
+   - Tailor tone based on 'style' (formal or informal).
+   - Handle errors (e.g., 400, 403, 429) with explanations (e.g., '403: The website may restrict scraping; try a different URL').
+   - Return the raw JSON response from the tools for search and analysis queries.
+   - Use 'history' to interpret context from previous queries.
+   - Respond in the specified 'language'.
+
+WEB_READING_GUIDE:
+{WEB_READING_GUIDE}
+            """,
+            inputs=["query", "style", "history", "language"],
+            outputs=[
+                PromptOutput(
+                    key="response",
+                    description="Raw JSON response from the search or scraping tool, or a conversation answer in plain text."
+                )
+            ],
+            examples=[
+                PromptExample(
+                    input_data="How do I identify credible news sources? [informal] [] en",
+                    output="Look for bylines, recent publication dates, and reputable domains like .edu or .gov. Avoid sites with sensational headlines or no clear author."
+                ),
+                PromptExample(
+                    input_data="Find recent tech news [formal] [] en",
+                    output=json.dumps({
+                        "results": [
+                            {"url": "https://techsite1.com", "title": "AI Breakthrough in 2025", "snippet": "New AI model achieves record accuracy..."},
+                            {"url": "https://techsite2.com", "title": "Quantum Computing Update", "snippet": "Advancements in quantum tech..."}
+                        ]
+                    })
+                ),
+                PromptExample(
+                    input_data="Summarize the news article at https://example.com/news [formal] [] en",
+                    output=json.dumps({
+                        "html": "<html><body><h1>New Product Launch</h1><p>Company X unveiled a new gadget on May 25, 2025, targeting the Y market...</p></body></html>",
+                        "css": "",
+                        "js": ""
+                    })
+                ),
+                PromptExample(
+                    input_data="What’s the main point of https://example.com/blog [formal] [] en",
+                    output=json.dumps({
+                        "html": "<html><body><article><h1>Why UX Matters</h1><p>User experience drives engagement and retention...</p></article></body></html>",
+                        "css": "",
+                        "js": ""
+                    })
+                ),
+                PromptExample(
+                    input_data="Explain the content of https://example.com [formal] [] en",
+                    output=json.dumps({
+                        "html": "<html><body><h1>Welcome to Example</h1><p>This site showcases our products and services...</p></body></html>",
+                        "css": "",
+                        "js": ""
+                    })
+                )
+            ]
+        ),
+        llm_config=LlmConfig(
+            max_tokens=5000,
+            timeout=0,
+            sampling=Sampling(temperature=0.7, top_k=50, top_p=1.0)
+        ),
+        models=ModelList(models=[Model(name="openai/gpt-4.1")]),
+        resource_pools=ResourcePoolList(
+            resource_pools=[
+                ResourcePool(
+                    name="WebReadingTools",
+                    tools=[
+                        ResourcePoolTool(name="get_search_api_v1_web_search__get_get", revision=None),
+                        ResourcePoolTool(name="tool_web_scrap_httpx_post", revision=None)
+                    ],
+                    agents=[]
+                )
+            ]
+        )
+    )
+)
+
+# Create the agent
+manager = AILabManager()
+result = manager.create_agent(
+    project_id="2ca6883f-6778-40bb-bcc1-85451fb11107",
+    agent=agent,
+    automatic_publish=False
+)
+
+if isinstance(result, Agent):
+    print(f"Agent created successfully: {agent.to_dict()}")
+else:
+    print("Errors:", result.errors if hasattr(result, 'errors') else "Unknown error occurred")

pygeai/tests/snippets/lab/use_cases/file_summarizer_example.py

@@ -0,0 +1,157 @@
+from uuid import uuid4
+
+from pygeai.chat.managers import ChatManager
+from pygeai.core.files.responses import UploadFileResponse
+from pygeai.lab.managers import AILabManager
+from pygeai.core.files.managers import FileManager
+from pygeai.core.files.models import UploadFile
+from pygeai.core.models import ChatMessageList, ChatMessage, LlmSettings
+from pygeai.lab.models import Agent, AgentData, Prompt, PromptExample, PromptOutput, LlmConfig, Sampling, Model, ModelList
+
+project_id = "2ca6883f-6778-40bb-bcc1-85451fb11107"
+
+lab_manager = AILabManager()
+file_manager = FileManager(
+    organization_id="4aa15b61-d3c7-4a5c-99b8-052d18a04ff2",
+    project_id=project_id
+)
+chat_manager = ChatManager()
+
+created_entities = {
+    "agent_id": None,
+    "file_id": None
+}
+
+
+def rollback():
+    print("\n=== Initiating Rollback ===")
+    if created_entities["agent_id"]:
+        print(f"Deleting agent {created_entities['agent_id']}...")
+        result = lab_manager.delete_agent(agent_id=created_entities["agent_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()
+
+    # Agent Flow
+    print("\n=== Agent Flow ===")
+    print("Creating agent 'FileSummaryAgent' as draft...")
+    agent_id = str(uuid4())
+    agent = Agent(
+        id=agent_id,
+        name="FileSummaryAgent",
+        access_scope="private",
+        job_description="Reads and summarizes content from text files",
+        description="An agent designed to process and summarize file content",
+        agent_data=AgentData(
+            prompt=Prompt(
+                instructions="Read the provided text file by its ID and generate a concise summary.",
+                inputs=["file_id"],
+                outputs=[
+                    PromptOutput(key="summary", description="Summary of the file content")
+                ],
+                examples=[
+                    PromptExample(
+                        input_data=f"File ID: {created_entities['file_id']}",
+                        output='{"summary": "Summary of test.txt content."}'
+                    )
+                ]
+            ),
+            llm_config=LlmConfig(
+                max_tokens=2000,
+                timeout=60,
+                sampling=Sampling(temperature=0.8)
+            ),
+            models=ModelList(models=[
+                Model(name="gpt-4-turbo")
+            ])
+        ),
+        is_draft=True,
+        revision=1,
+        status="pending"
+    )
+    create_agent_result = lab_manager.create_agent(agent=agent, automatic_publish=False)
+    if isinstance(create_agent_result, Agent):
+        print(f"Success: Created Agent: {create_agent_result.name}, ID: {create_agent_result.id}")
+        created_entities["agent_id"] = create_agent_result.id
+    else:
+        print("Error: Agent creation failed:", create_agent_result)
+        rollback()
+        exit()
+
+    print("Publishing agent revision '1'...")
+    publish_agent_result = lab_manager.publish_agent_revision(agent_id=created_entities["agent_id"], revision="1")
+    if isinstance(publish_agent_result, Agent):
+        print(f"Success: Published Agent Revision: {publish_agent_result.name}")
+    else:
+        print("Error: Agent publish failed:", publish_agent_result)
+        rollback()
+        exit()
+
+    # Chat Completion Flow
+    print("\n=== Chat Completion Flow ===")
+    print(f"Asking agent 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."
+        )
+    ])
+    llm_settings = LlmSettings(
+        temperature=0.7,
+        max_tokens=500
+    )
+    chat_response = chat_manager.chat_completion(
+        model=f"saia:agent:{agent.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}")