PraisonAI 2.0.61__cp313-cp313-manylinux_2_39_x86_64.whl

praisonai/ui/config/translations/zh-CN.json

@@ -0,0 +1,229 @@
+{
+    "components": {
+        "atoms": {
+            "buttons": {
+                "userButton": {
+                    "menu": {
+                        "settings": "\u8bbe\u7f6e",
+                        "settingsKey": "S",
+                        "APIKeys": "API \u5bc6\u94a5",
+                        "logout": "\u767b\u51fa"
+                    }
+                }
+            }
+        },
+        "molecules": {
+            "newChatButton": {
+                "newChat": "\u65b0\u5efa\u5bf9\u8bdd"
+            },
+            "tasklist": {
+                "TaskList": {
+                    "title": "\ud83d\uddd2\ufe0f \u4efb\u52a1\u5217\u8868",
+                    "loading": "\u52a0\u8f7d\u4e2d...",
+                    "error": "\u53d1\u751f\u9519\u8bef"
+                }
+            },
+            "attachments": {
+                "cancelUpload": "\u53d6\u6d88\u4e0a\u4f20",
+                "removeAttachment": "\u79fb\u9664\u9644\u4ef6"
+            },
+            "newChatDialog": {
+                "createNewChat": "\u521b\u5efa\u65b0\u5bf9\u8bdd\uff1f",
+                "clearChat": "\u8fd9\u5c06\u6e05\u9664\u5f53\u524d\u6d88\u606f\u5e76\u5f00\u59cb\u65b0\u7684\u5bf9\u8bdd\u3002",
+                "cancel": "\u53d6\u6d88",
+                "confirm": "\u786e\u8ba4"
+            },
+            "settingsModal": {
+                "settings": "\u8bbe\u7f6e",
+                "expandMessages": "\u5c55\u5f00\u6d88\u606f",
+                "hideChainOfThought": "\u9690\u85cf\u601d\u8003\u94fe",
+                "darkMode": "\u6697\u8272\u6a21\u5f0f"
+            },
+            "detailsButton": {
+                "using": "\u4f7f\u7528",
+                "used": "\u5df2\u7528"
+            },
+            "auth": {
+                "authLogin": {
+                    "title": "\u767b\u5f55\u4ee5\u8bbf\u95ee\u5e94\u7528\u3002",
+                    "form": {
+                        "email": "\u7535\u5b50\u90ae\u7bb1\u5730\u5740",
+                        "password": "\u5bc6\u7801",
+                        "noAccount": "\u6ca1\u6709\u8d26\u6237\uff1f",
+                        "alreadyHaveAccount": "\u5df2\u6709\u8d26\u6237\uff1f",
+                        "signup": "\u6ce8\u518c",
+                        "signin": "\u767b\u5f55",
+                        "or": "\u6216\u8005",
+                        "continue": "\u7ee7\u7eed",
+                        "forgotPassword": "\u5fd8\u8bb0\u5bc6\u7801\uff1f",
+                        "passwordMustContain": "\u60a8\u7684\u5bc6\u7801\u5fc5\u987b\u5305\u542b\uff1a",
+                        "emailRequired": "\u7535\u5b50\u90ae\u7bb1\u662f\u5fc5\u586b\u9879",
+                        "passwordRequired": "\u5bc6\u7801\u662f\u5fc5\u586b\u9879"
+                    },
+                    "error": {
+                        "default": "\u65e0\u6cd5\u767b\u5f55\u3002",
+                        "signin": "\u5c1d\u8bd5\u4f7f\u7528\u4e0d\u540c\u7684\u8d26\u6237\u767b\u5f55\u3002",
+                        "oauthsignin": "\u5c1d\u8bd5\u4f7f\u7528\u4e0d\u540c\u7684\u8d26\u6237\u767b\u5f55\u3002",
+                        "redirect_uri_mismatch": "\u91cd\u5b9a\u5411URI\u4e0eOAuth\u5e94\u7528\u914d\u7f6e\u4e0d\u5339\u914d\u3002",
+                        "oauthcallbackerror": "\u5c1d\u8bd5\u4f7f\u7528\u4e0d\u540c\u7684\u8d26\u6237\u767b\u5f55\u3002",
+                        "oauthcreateaccount": "\u5c1d\u8bd5\u4f7f\u7528\u4e0d\u540c\u7684\u8d26\u6237\u767b\u5f55\u3002",
+                        "emailcreateaccount": "\u5c1d\u8bd5\u4f7f\u7528\u4e0d\u540c\u7684\u8d26\u6237\u767b\u5f55\u3002",
+                        "callback": "\u5c1d\u8bd5\u4f7f\u7528\u4e0d\u540c\u7684\u8d26\u6237\u767b\u5f55\u3002",
+                        "oauthaccountnotlinked": "\u4e3a\u4e86\u9a8c\u8bc1\u60a8\u7684\u8eab\u4efd\uff0c\u8bf7\u4f7f\u7528\u6700\u521d\u4f7f\u7528\u7684\u540c\u4e00\u8d26\u6237\u767b\u5f55\u3002",
+                        "emailsignin": "\u65e0\u6cd5\u53d1\u9001\u90ae\u4ef6\u3002",
+                        "emailverify": "\u8bf7\u9a8c\u8bc1\u60a8\u7684\u7535\u5b50\u90ae\u4ef6\uff0c\u5df2\u53d1\u9001\u4e00\u5c01\u65b0\u90ae\u4ef6\u3002",
+                        "credentialssignin": "\u767b\u5f55\u5931\u8d25\u3002\u8bf7\u68c0\u67e5\u60a8\u63d0\u4f9b\u7684\u8be6\u7ec6\u4fe1\u606f\u662f\u5426\u6b63\u786e\u3002",
+                        "sessionrequired": "\u8bf7\u767b\u5f55\u4ee5\u8bbf\u95ee\u6b64\u9875\u9762\u3002"
+                    }
+                },
+                "authVerifyEmail": {
+                    "almostThere": "\u60a8\u5feb\u6210\u529f\u4e86\uff01\u6211\u4eec\u5df2\u5411 ",
+                    "verifyEmailLink": "\u8bf7\u5355\u51fb\u8be5\u90ae\u4ef6\u4e2d\u7684\u94fe\u63a5\u4ee5\u5b8c\u6210\u6ce8\u518c\u3002",
+                    "didNotReceive": "\u6ca1\u627e\u5230\u90ae\u4ef6\uff1f",
+                    "resendEmail": "\u91cd\u65b0\u53d1\u9001\u90ae\u4ef6",
+                    "goBack": "\u8fd4\u56de",
+                    "emailSent": "\u90ae\u4ef6\u5df2\u6210\u529f\u53d1\u9001\u3002",
+                    "verifyEmail": "\u9a8c\u8bc1\u60a8\u7684\u7535\u5b50\u90ae\u4ef6\u5730\u5740"
+                },
+                "providerButton": {
+                    "continue": "\u4f7f\u7528{{provider}}\u7ee7\u7eed",
+                    "signup": "\u4f7f\u7528{{provider}}\u6ce8\u518c"
+                },
+                "authResetPassword": {
+                    "newPasswordRequired": "\u65b0\u5bc6\u7801\u662f\u5fc5\u586b\u9879",
+                    "passwordsMustMatch": "\u5bc6\u7801\u5fc5\u987b\u4e00\u81f4",
+                    "confirmPasswordRequired": "\u786e\u8ba4\u5bc6\u7801\u662f\u5fc5\u586b\u9879",
+                    "newPassword": "\u65b0\u5bc6\u7801",
+                    "confirmPassword": "\u786e\u8ba4\u5bc6\u7801",
+                    "resetPassword": "\u91cd\u7f6e\u5bc6\u7801"
+                },
+                "authForgotPassword": {
+                    "email": "\u7535\u5b50\u90ae\u7bb1\u5730\u5740",
+                    "emailRequired": "\u7535\u5b50\u90ae\u7bb1\u662f\u5fc5\u586b\u9879",
+                    "emailSent": "\u8bf7\u68c0\u67e5\u7535\u5b50\u90ae\u7bb1{{email}}\u4ee5\u83b7\u53d6\u91cd\u7f6e\u5bc6\u7801\u7684\u6307\u793a\u3002",
+                    "enterEmail": "\u8bf7\u8f93\u5165\u60a8\u7684\u7535\u5b50\u90ae\u7bb1\u5730\u5740\uff0c\u6211\u4eec\u5c06\u53d1\u9001\u91cd\u7f6e\u5bc6\u7801\u7684\u6307\u793a\u3002",
+                    "resendEmail": "\u91cd\u65b0\u53d1\u9001\u90ae\u4ef6",
+                    "continue": "\u7ee7\u7eed",
+                    "goBack": "\u8fd4\u56de"
+                }
+            }
+        },
+        "organisms": {
+            "chat": {
+                "history": {
+                    "index": {
+                        "showHistory": "\u663e\u793a\u5386\u53f2",
+                        "lastInputs": "\u6700\u540e\u8f93\u5165",
+                        "noInputs": "\u5982\u6b64\u7a7a\u65f7...",
+                        "loading": "\u52a0\u8f7d\u4e2d..."
+                    }
+                },
+                "inputBox": {
+                    "input": {
+                        "placeholder": "\u5728\u8fd9\u91cc\u8f93\u5165\u60a8\u7684\u6d88\u606f..."
+                    },
+                    "speechButton": {
+                        "start": "\u5f00\u59cb\u5f55\u97f3",
+                        "stop": "\u505c\u6b62\u5f55\u97f3"
+                    },
+                    "SubmitButton": {
+                        "sendMessage": "\u53d1\u9001\u6d88\u606f",
+                        "stopTask": "\u505c\u6b62\u4efb\u52a1"
+                    },
+                    "UploadButton": {
+                        "attachFiles": "\u9644\u52a0\u6587\u4ef6"
+                    },
+                    "waterMark": {
+                        "text": "\u4f7f\u7528"
+                    }
+                },
+                "Messages": {
+                    "index": {
+                        "running": "\u8fd0\u884c\u4e2d",
+                        "executedSuccessfully": "\u6267\u884c\u6210\u529f",
+                        "failed": "\u5931\u8d25",
+                        "feedbackUpdated": "\u53cd\u9988\u66f4\u65b0",
+                        "updating": "\u6b63\u5728\u66f4\u65b0"
+                    }
+                },
+                "dropScreen": {
+                    "dropYourFilesHere": "\u5728\u8fd9\u91cc\u62d6\u653e\u60a8\u7684\u6587\u4ef6"
+                },
+                "index": {
+                    "failedToUpload": "\u4e0a\u4f20\u5931\u8d25",
+                    "cancelledUploadOf": "\u53d6\u6d88\u4e0a\u4f20",
+                    "couldNotReachServer": "\u65e0\u6cd5\u8fde\u63a5\u5230\u670d\u52a1\u5668",
+                    "continuingChat": "\u7ee7\u7eed\u4e4b\u524d\u7684\u5bf9\u8bdd"
+                },
+                "settings": {
+                    "settingsPanel": "\u8bbe\u7f6e\u9762\u677f",
+                    "reset": "\u91cd\u7f6e",
+                    "cancel": "\u53d6\u6d88",
+                    "confirm": "\u786e\u8ba4"
+                }
+            },
+            "threadHistory": {
+                "sidebar": {
+                    "filters": {
+                        "FeedbackSelect": {
+                            "feedbackAll": "\u53cd\u9988\uff1a\u5168\u90e8",
+                            "feedbackPositive": "\u53cd\u9988\uff1a\u6b63\u9762",
+                            "feedbackNegative": "\u53cd\u9988\uff1a\u8d1f\u9762"
+                        },
+                        "SearchBar": {
+                            "search": "\u641c\u7d22"
+                        }
+                    },
+                    "DeleteThreadButton": {
+                        "confirmMessage": "\u8fd9\u5c06\u5220\u9664\u7ebf\u7a0b\u53ca\u5176\u6d88\u606f\u548c\u5143\u7d20\u3002",
+                        "cancel": "\u53d6\u6d88",
+                        "confirm": "\u786e\u8ba4",
+                        "deletingChat": "\u5220\u9664\u5bf9\u8bdd",
+                        "chatDeleted": "\u5bf9\u8bdd\u5df2\u5220\u9664"
+                    },
+                    "index": {
+                        "pastChats": "\u8fc7\u5f80\u5bf9\u8bdd"
+                    },
+                    "ThreadList": {
+                        "empty": "\u7a7a\u7684...",
+                        "today": "\u4eca\u5929",
+                        "yesterday": "\u6628\u5929",
+                        "previous7days": "\u524d7\u5929",
+                        "previous30days": "\u524d30\u5929"
+                    },
+                    "TriggerButton": {
+                        "closeSidebar": "\u5173\u95ed\u4fa7\u8fb9\u680f",
+                        "openSidebar": "\u6253\u5f00\u4fa7\u8fb9\u680f"
+                    }
+                },
+                "Thread": {
+                    "backToChat": "\u8fd4\u56de\u5bf9\u8bdd",
+                    "chatCreatedOn": "\u6b64\u5bf9\u8bdd\u521b\u5efa\u4e8e"
+                }
+            },
+            "header": {
+                "chat": "\u5bf9\u8bdd",
+                "readme": "\u8bf4\u660e"
+            }
+        }
+    },
+    "hooks": {
+        "useLLMProviders": {
+            "failedToFetchProviders": "\u83b7\u53d6\u63d0\u4f9b\u8005\u5931\u8d25:"
+        }
+    },
+    "pages": {
+        "Design": {},
+        "Env": {
+            "savedSuccessfully": "\u4fdd\u5b58\u6210\u529f",
+            "requiredApiKeys": "\u5fc5\u9700\u7684API\u5bc6\u94a5",
+            "requiredApiKeysInfo": "\u8981\u4f7f\u7528\u6b64\u5e94\u7528\uff0c\u9700\u8981\u4ee5\u4e0bAPI\u5bc6\u94a5\u3002\u8fd9\u4e9b\u5bc6\u94a5\u5b58\u50a8\u5728\u60a8\u7684\u8bbe\u5907\u672c\u5730\u5b58\u50a8\u4e2d\u3002"
+        },
+        "Page": {
+            "notPartOfProject": "\u60a8\u4e0d\u662f\u6b64\u9879\u76ee\u7684\u4e00\u90e8\u5206\u3002"
+        },
+        "ResumeButton": {
+            "resumeChat": "\u6062\u590d\u5bf9\u8bdd"
+        }
+    }
+}

praisonai/ui/context.py

@@ -0,0 +1,283 @@
+import os
+import fnmatch
+import yaml
+from pathlib import Path
+import logging
+
+# Set up logging
+logger = logging.getLogger(__name__)
+log_level = os.getenv("LOGLEVEL", "INFO").upper()
+logger.handlers = []
+
+# Set up logging to console
+console_handler = logging.StreamHandler()
+console_handler.setLevel(log_level)
+console_formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+console_handler.setFormatter(console_formatter)
+logger.addHandler(console_handler)
+
+# Set the logging level for the logger
+logger.setLevel(log_level)
+
+class ContextGatherer:
+    def __init__(self, directory='.', output_file='context.txt',
+                 relevant_extensions=None, max_file_size=1_000_000, max_tokens=900000):
+        self.directory = directory
+        self.output_file = output_file
+        self.relevant_extensions = relevant_extensions or [
+            '.py', '.js', '.ts', '.java', '.rb', '.php', '.pl', '.pm', '.c', '.h',
+            '.cpp', '.hpp', '.cs', '.vb', '.swift', '.kt', '.m', '.mm', '.go', '.rs',
+            '.hs', '.r', '.lua', '.sh', '.bat', '.clj', '.scala', '.erl', '.ex',
+            '.ml', '.fs', '.groovy', '.jsm', '.jsx', '.tsx', '.yaml'
+        ]
+        self.max_file_size = max_file_size
+        self.max_tokens = int(os.getenv("PRAISONAI_MAX_TOKENS", max_tokens))
+        self.ignore_patterns = self.get_ignore_patterns()
+        self.include_paths = self.get_include_paths()
+        self.included_files = []
+
+    def get_ignore_patterns(self):
+        """
+        Loads ignore patterns from various sources, prioritizing them in
+        the following order:
+            1. .praisonignore
+            2. settings.yaml (under code.ignore_files)
+            3. PRAISONAI_IGNORE_FILES environment variable
+            4. .gitignore
+            5. Default patterns
+        """
+        ignore_patterns = []
+
+        def load_from_file(filepath):
+            if os.path.exists(filepath):
+                with open(filepath, 'r') as f:
+                    ignore_patterns.extend(
+                        line.strip() for line in f 
+                        if line.strip() and not line.startswith('#')
+                    )
+
+        # 1. Load from .praisonignore
+        load_from_file(os.path.join(self.directory, '.praisonignore'))
+
+        # 2. Load from settings.yaml
+        settings_path = os.path.join(self.directory, 'settings.yaml')
+        if os.path.exists(settings_path):
+            with open(settings_path, 'r') as f:
+                settings = yaml.safe_load(f)
+                if 'code' in settings and 'ignore_files' in settings['code']:
+                    ignore_patterns.extend(settings['code']['ignore_files'])
+
+        # 3. Load from environment variable
+        ignore_files_env = os.getenv("PRAISONAI_IGNORE_FILES")
+        if ignore_files_env:
+            ignore_patterns.extend(ignore_files_env.split(","))
+
+        # 4. Load from .gitignore
+        load_from_file(os.path.join(self.directory, '.gitignore'))
+
+        # 5. Default patterns (only if no patterns loaded from above sources)
+        if not ignore_patterns:
+            ignore_patterns = [
+                ".*", "*.pyc", "__pycache__", ".git", ".gitignore", ".vscode",
+                ".idea", ".DS_Store", "*.lock", "*.pyc", ".env", "docs", "tests", 
+                "test", "tmp", "temp", "*.txt", "*.md", "*.json", "*.csv", "*.tsv",
+                "public", "*.sql", "*.sqlite", "*.db", "*.db3", "*.sqlite3", 
+                "*.log", "*.zip", "*.gz", "*.tar", "*.rar", "*.7z", "*.pdf", 
+                "*.jpg", "*.jpeg", "*.png", "*.gif", "*.svg", "cookbooks", 
+                "assets", "__pycache__", "dist", "build", "node_modules", "venv"
+            ]
+            logger.debug(f"Using default ignore patterns: {ignore_patterns}")
+
+        # Modify patterns to match directories and add leading '*' if necessary
+        modified_ignore_patterns = [
+            '*' + pattern if not pattern.startswith('.') and not pattern.startswith('*') else pattern
+            for pattern in ignore_patterns
+        ]
+        logger.debug(f"Final ignore patterns: {modified_ignore_patterns}")
+        return modified_ignore_patterns
+
+    def get_include_paths(self):
+        """
+        Loads include paths from:
+            1. .praisoninclude (includes ONLY files/directories listed)
+            2. .praisoncontext (if .praisoninclude doesn't exist, this is used
+               to include all other relevant files, excluding ignore patterns)
+        """
+        include_paths = []
+        include_all = False  # Flag to indicate if we need to include all files
+ 
+        include_file = os.path.join(self.directory, '.praisoncontext')
+        if os.path.exists(include_file):
+            with open(include_file, 'r') as f:
+                include_paths.extend(
+                    line.strip() for line in f
+                    if line.strip() and not line.startswith('#')
+                )
+ 
+        # If .praisoncontext doesn't exist, fall back to .praisoninclude
+        # for including all relevant files
+        if not include_paths: 
+            include_file = os.path.join(self.directory, '.praisoninclude')
+            if os.path.exists(include_file):
+                with open(include_file, 'r') as f:
+                    include_paths.extend(
+                        line.strip() for line in f
+                        if line.strip() and not line.startswith('#')
+                    )
+                include_all = True  # Include all files along with specified paths
+ 
+        return include_paths, include_all
+
+    def should_ignore(self, file_path):
+        """
+        Check if a file or directory should be ignored based on patterns.
+        Handles both file names and directory names for more comprehensive filtering.
+        """
+        relative_path = os.path.relpath(file_path, self.directory)
+        if relative_path.startswith('.'):
+            return True
+        for pattern in self.ignore_patterns:
+            if fnmatch.fnmatch(relative_path, pattern) or \
+               fnmatch.fnmatch(os.path.basename(file_path), pattern):
+                return True
+        return False
+
+    def is_relevant_file(self, file_path):
+        """Determine if a file is relevant for the context."""
+        return os.path.isfile(file_path) and \
+               os.path.getsize(file_path) <= self.max_file_size and \
+               any(file_path.endswith(ext) for ext in self.relevant_extensions)
+
+    def gather_context(self):
+        """
+        Gather context from relevant files, respecting ignore patterns
+        and include options from .praisoninclude and .praisoncontext.
+        """
+        context = []
+        total_files = 0
+        processed_files = 0
+        self.include_paths, include_all = self.get_include_paths()
+
+        def add_file_content(file_path):
+            """Helper function to add file content to context."""
+            try:
+                with open(file_path, 'r', encoding='utf-8') as f:
+                    content = f.read()
+                    context.append(
+                        f"File: {file_path}\n\n{content}\n\n{'=' * 50}\n"
+                    )
+                    self.included_files.append(
+                        Path(file_path).relative_to(self.directory)
+                    )
+            except Exception as e:
+                logger.error(f"Error reading {file_path}: {e}")
+
+        def process_path(path):
+            """Helper function to process a single path (file or directory)."""
+            nonlocal total_files, processed_files
+            if os.path.isdir(path):
+                for root, dirs, files in os.walk(path):
+                    total_files += len(files)
+                    dirs[:] = [
+                        d
+                        for d in dirs
+                        if not self.should_ignore(os.path.join(root, d))
+                    ]
+                    for file in files:
+                        file_path = os.path.join(root, file)
+                        if not self.should_ignore(file_path) and self.is_relevant_file(file_path):
+                            add_file_content(file_path)
+                        processed_files += 1
+                        print(
+                            f"\rProcessed {processed_files}/{total_files} files",
+                            end="",
+                            flush=True,
+                        )
+            elif os.path.isfile(path) and self.is_relevant_file(path):
+                add_file_content(path)
+                processed_files += 1
+                print(
+                    f"\rProcessed {processed_files}/1 files",
+                    end="",
+                    flush=True,
+                )
+
+        if include_all:
+            # Include ALL relevant files from the entire directory
+            process_path(self.directory)
+            
+            # Include files from .praisoninclude specifically
+            for include_path in self.include_paths:
+                full_path = os.path.join(self.directory, include_path)
+                process_path(full_path)
+        elif self.include_paths:
+            # Include only files specified in .praisoncontext
+            for include_path in self.include_paths:
+                full_path = os.path.join(self.directory, include_path)
+                process_path(full_path)
+        else:
+            # No include options, process the entire directory
+            process_path(self.directory)
+
+        print()  # New line after progress indicator
+        return "\n".join(context)
+
+    def count_tokens(self, text):
+        """Count tokens using a simple whitespace-based tokenizer."""
+        return len(text.split())
+
+    def truncate_context(self, context):
+        """Truncate context to stay within the token limit."""
+        tokens = context.split()
+        if len(tokens) > self.max_tokens:
+            truncated_context = ' '.join(tokens[:self.max_tokens])
+            logger.warning("Context truncated due to token limit.")
+            return truncated_context
+        return context
+
+    def save_context(self, context):
+        """Save the gathered context to a file."""
+        with open(self.output_file, 'w', encoding='utf-8') as f:
+            f.write(context)
+
+    def get_context_tree(self):
+        """Generate a formatted tree structure of included files and folders."""
+        tree = []
+        start_dir = Path(self.directory)
+
+        def add_to_tree(path, prefix=''):
+            contents = sorted(path.iterdir())
+            pointers = [('└── ' if i == len(contents) - 1 else '├── ') for i in range(len(contents))]
+            for pointer, item in zip(pointers, contents):
+                rel_path = item.relative_to(start_dir)
+                if rel_path in self.included_files:
+                    tree.append(f"{prefix}{pointer}{rel_path}")
+
+                if item.is_dir():
+                    add_to_tree(item, prefix + ('    ' if pointer == '└── ' else '│   '))
+
+        add_to_tree(start_dir)
+        return '\n'.join(tree)
+
+    def run(self):
+        """Execute the context gathering, truncation, and reporting."""
+        context = self.gather_context()
+        context = self.truncate_context(context)
+        token_count = self.count_tokens(context)
+        print(f"Context gathered successfully.")
+        print(f"Total number of tokens (estimated): {token_count}")
+        # self.save_context(context)
+        context_tree = self.get_context_tree()
+        logger.debug(f"Context tree:\n{context_tree}")
+        return context, token_count, context_tree
+
+def main():
+    gatherer = ContextGatherer()
+    context, token_count, context_tree = gatherer.run()
+    print(context_tree)
+    print(f"\nThe context contains approximately {token_count} tokens.")
+    print("First 500 characters of context:")
+    print(context[:500] + "...")
+
+if __name__ == "__main__":
+    main()