U h@ @sddlmZmZmZmZddlmZmZddlm Z ddl m Z ddl m Z ddlZddlZddlmZmZmZmZejejdd eeZe d d Zed d ZeddZeddeeeefeeefeedddZeedddZ edeeeeeeeedddZ!dS))capture_exceptioncapture_messageset_tag set_context)monitor_openai_apimonitor_critical_system) ChatOpenAI)LangChainTracer)OpenAIN)OptionalListDictAnyz4%(asctime)s - %(name)s - %(levelname)s - %(message)s)levelformatzopenai-cost-tracking) project_namec Cs^zddlm}|WStk rX}z(td|t|ddiWYdSd}~XYnXdS)z Get OpenAI API key from settingsr)OPENAI_API_KEYzError loading OpenAI API key: functionget_openai_api_keyN)config.settingsr Exceptionloggererrorr)rer9/var/www/html/testcasegenerator.evdpl.com/ai/generator.pyrs rc Csnz"ddlm}m}m}|||dWStk rh}z(td|t|ddiWYdSd}~XYnXdS)z*Get OpenRouter configuration from settingsr)OPENROUTER_API_KEYOPENROUTER_SITE_URLOPENROUTER_SITE_NAME)api_keysite_url site_namez(Error loading OpenRouter configuration: rget_openrouter_configN)rrrrrrrr)rrrrrrrr"sr"Tcritical)messagesconfigreturnc Cszb|r|ds tdWdStd|dd}|jjj|d|ddid |d }|jd jj WSt k r}z(td |t |d diWYdSd}~XYnXdS)z'Make API call to OpenRouter as fallbackrz OpenRouter API key not availableNzhttps://openrouter.ai/api/v1)base_urlrr r!)z HTTP-RefererzX-Titlez deepseek/deepseek-chat-v3.1:free)Z extra_headersZ extra_bodymodelr%rz"Error making OpenRouter API call: rmake_openrouter_call) getrrr ZchatZ completionscreatechoicesmessagecontentrr)r%r&clientZ completionrrrrr*-s*  r*) test_typer'c CsNddddddddddd dd d d dd d d dddd dd}||iS)z.Get the configuration for a specific test typeZTC_FUNCzEfunctional test cases focusing on valid inputs and expected behaviors)prefix description max_countZTC_NEGzNnegative test cases focusing on invalid inputs, error handling, and edge casesZTC_UIz4UI test cases focusing on visual elements and layoutZTC_UXzDuser experience test cases focusing on user interaction and workflowZ TC_COMPATz@compatibility test cases across different browsers and platformsZTC_PERFz@performance test cases focusing on load times and responsiveness)dashboard_functionalZdashboard_negativeZ dashboard_uiZ dashboard_uxZdashboard_compatibilityZdashboard_performance)r+)r1Z base_configsrrrget_test_type_configKs4 r8)r4summaryselected_types source_typeurlr'cs|stddS|r"t|dkr0tddStd|td|tdt|dg}|D](td tstd qnd d d }|dkrd|d|dddddd dd|dnd%td?Wnttk rt}zTtd@d0t|t| ||rFt|ndt|dAWY WYqnW5d}~XYnXW5d} ~ XYnXtdBqn|stdCdStdDt|dEd-d-|S)Fz0Generate test cases based on user-selected typesz0No description provided for test case generationNrz/No test types selected for test case generationz!Generating test cases for types: z Summary: zDescription length: z charactersz#Starting generation for test type: zSkipping unknown test type: u You are an expert QA engineer. Your task is to create **clear, detailed, and professional test cases**. ✅ STRICT FORMAT for each test case: Title: r3u_[SequentialNumber]_[Meaningful_Brief_Title] Scenario: [One or two lines describing the intent of the test] Preconditions: [State assumptions, setup, or required data before execution] Steps to Reproduce: 1. [Step 1 in action-oriented language] 2. [Step 2] ... Expected Result: [Clear, testable outcome of the steps] Actual Result: [Leave as 'To be filled during execution'] Priority: [High / Medium / Low] Test Data: [If applicable – specify input values, files, or environment details] ⚡ RULES: - Each test case must cover a **unique and valuable scenario**. - Steps must be **clear, actionable, and written like instructions to a QA engineer**. - Expected Results must be **specific and measurable** (not vague like “It should work”). - Do NOT duplicate scenarios. - Do NOT mix with other test types. - Use professional QA terminology. r=z Website URL: z" Content Description: z Generate **up to r5z test cases** for r4u. ⚡ ANALYSIS REQUIREMENTS: - Carefully review the website’s structure and functionality - Think about real-world user workflows and how they may fail - Consider UI, UX, and interaction edge cases - Include cross-browser compatibility (Chrome, Firefox, Safari, Edge) - Consider mobile responsiveness (desktop vs. mobile behavior) ⚡ TEST CASE REQUIREMENTS: 1. Prefix all test case titles with z" 2. Focus **only** on a test scenarios for web testing 3. Write **detailed, step-by-step instructions** that can be executed by QA 4. Specify **exact expected results** for every step or final outcome 5. Ensure scenarios are **unique and non-overlapping** 6. Include both **happy paths and edge cases** 7. Add **test data** where necessary z z Task Title: z Task Description: u. ⚡ ANALYSIS REQUIREMENTS: - Carefully analyze the task description - Identify **all functional components and user actions** - Think about positive, negative, and edge case scenarios - Consider different **user roles, permissions, and data inputs** - Anticipate possible **failure points** or boundary conditions ⚡ TEST CASE REQUIREMENTS: 1. Prefix all test case titles with aQ scenarios 3. Write **step-by-step reproducible instructions** 4. Provide **clear and measurable expected results** 5. Ensure each case is **unique and does not overlap** 6. Include **test data** where relevant 7. Include **priority level** for execution planning your_openai_api_key_heremissing_api_keyuA⚠️ Invalid or missing OPENAI_API_KEY in environment variablesz gpt-4o-minig333333?i.)r)Z temperature max_tokensZopenai_api_key callbackszSending request to OpenAI for z test casesTr#c s6ddddddddd dgS) NsystemAYou are a senior QA engineer. Generate the appropriate number of  test cases (up to r5[ maximum) by analyzing the content complexity and generating only what's truly needed. Use r3= as the prefix. Focus on quality and relevance over quantity.roler/user)invokerr&Z current_llmpromptr1rrmake_openai_call2s z,generate_test_case..make_openai_callz Generated z% test cases successfully using OpenAIz TEST TYPE: z zReceived empty response for zOpenAI API failed for z test cases: z&Attempting fallback to OpenRouter for rz3OpenRouter configuration not available for fallbackzNo fallback API availablerBrCrDrErFrGrIz2 test cases successfully using OpenRouter fallbackz (Generated via OpenRouter) z$OpenRouter fallback also failed for z!Both OpenAI and OpenRouter failedz'Fallback to OpenRouter also failed for )r1r&r:Zdescription_lengthfallback_errorz$Completed generation for test type: z!Failed to generate any test casesz&Successfully generated test cases for z test types)rrleninfor8warningr ValueErrorrtracerrr/stripappendrstrr"r+r*rjoin)r4r:r;r<r=all_test_casesZ base_promptr error_msgrMresponse test_casesZtest_cases_with_headerrZopenrouter_configr%rNrrKrgenerate_test_caseos   K         2 r\)r9NNN)"utils.error_loggerrrrrutils.error_monitorrrZlangchain_openairZ%langchain.callbacks.tracers.langchainr openair osloggingtypingr r r r basicConfigINFO getLogger__name__rrSrr"rVr*dictr8r\rrrrs&       ,$