roksta 0.2.7__cp314-cp314-win_amd64.whl → 0.3.2__cp314-cp314-win_amd64.whl

{roksta-0.2.7.dist-info → roksta-0.3.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: roksta
-Version: 0.2.7
+Version: 0.3.2
 Summary: An AI coding assistant
 Author-email: Prash Naidu <prash@roksta.ai>
 Project-URL: Homepage, https://roksta.ai
@@ -21,6 +21,7 @@ Requires-Dist: gitpython>=3.1.44
 Requires-Dist: jsonref>=1.1.0
 Requires-Dist: pathspec>=0.12.1
 Requires-Dist: psutil>=7.1.0
+Requires-Dist: PyNaCl>=1.6.0
 Provides-Extra: tests
 Requires-Dist: pytest>=8.4.1; extra == "tests"
 Requires-Dist: pytest-asyncio>=1.0.0; extra == "tests"

roksta-0.3.2.dist-info/RECORD

@@ -0,0 +1,121 @@
+roksta/__init__.cp314-win_amd64.pyd,sha256=Y98BLa_mP5zxi3ECQTI5IZTpsvS3eVDB6nR5Ipz1_iI,16896
+roksta/analytics.cp314-win_amd64.pyd,sha256=4r6lQq2MQyjbQL7ns3zHeGvvMDdUPHNvhQ2jIcmGs2k,94208
+roksta/balance.cp314-win_amd64.pyd,sha256=L2qznmDduxRPbGZuvbCDv4kKhhY5ALkmgU8u6GLGitc,140288
+roksta/build_project.cp314-win_amd64.pyd,sha256=tV3YJZVOy5uoDY3UwV8rlArW4LWCyZcG0SbS_4MYEps,60928
+roksta/chat_workflow.cp314-win_amd64.pyd,sha256=C1Ymb9Sx79Gl-e3YdbxpTGvnT-USJhzX_hgsfu6SpGw,89600
+roksta/check_for_updates.cp314-win_amd64.pyd,sha256=FHL6S4igkEAxN_KpgPE_eyRw-VsaRcBfmO-FYX6eVog,102400
+roksta/checkpoints.cp314-win_amd64.pyd,sha256=CfGBzg3tNzaW3hgmIGgVWDMNFY4gsanMrUcrkl_O0Jo,79872
+roksta/clarify_goal.cp314-win_amd64.pyd,sha256=k1ZU8KtZxkMRIEUHlk0OaMH1nlsy-9j-EcgXkeyKweg,61440
+roksta/codebase_listing.cp314-win_amd64.pyd,sha256=Bgbgi6h0zHlZ9Ut_eSXInSQoD2WgUhaiosxuluXi8zs,47104
+roksta/create_default_config.cp314-win_amd64.pyd,sha256=dVv-RYH5cFNFRywJnc6DxEKqJzUV-30NMsovCZfbR3s,28672
+roksta/default_config.cp314-win_amd64.pyd,sha256=I6175fYVS-45XmFp90dt23q1SjKQbZnOTGcGBq51XFs,31744
+roksta/enums.cp314-win_amd64.pyd,sha256=YD4qjX3a3ETeQfUxvQ4PMNoLxdUATiXtW2ukd0WPxcQ,124416
+roksta/env.cp314-win_amd64.pyd,sha256=jx4mwJDKsjhDyUpv5LctmrUG4s7HNK-F6IM-kfNzYSg,26112
+roksta/extended_text_area.cp314-win_amd64.pyd,sha256=XUV39mWjTghPfyCsi75p8ewjnSYc4a2Fx2m4_2TTU98,105984
+roksta/firebase.cp314-win_amd64.pyd,sha256=6rxaxM4BdUiC7jvx9RORat54jAeDdO122sd2Sc5sGSA,187392
+roksta/firebase_auth_web.cp314-win_amd64.pyd,sha256=ySCMkZh4yse7jmUWn9w-pAZgjdsF7CdApKNM78esB_o,177152
+roksta/firebase_config.cp314-win_amd64.pyd,sha256=EGoI3qTE4vM8UaGe8ZtAgtUYNDdLNWa87nxGyyb-iNA,19968
+roksta/fix_tests.cp314-win_amd64.pyd,sha256=MuvU-vUYFP9Hx_i6YER4DyUzGYi_9ZNHFB-vih_FNw8,74240
+roksta/gen_codebase_summaries.cp314-win_amd64.pyd,sha256=aJQ7H0mcCoC7SsAFRx-UUb6crpPBX24dPeLlfCfowc4,243200
+roksta/gen_one_line_goal.cp314-win_amd64.pyd,sha256=C8f7YGpfdXyfe7FfRJcSed9mlyuI_b2Yq6fP66EuXa8,65024
+roksta/get_codebase_structure.cp314-win_amd64.pyd,sha256=9XN1As3je1RJ5nhMQJ5JCLGE_WysFoXlgIk-0MDRqDY,67072
+roksta/get_failing_tests.cp314-win_amd64.pyd,sha256=XpKsxsCezU-Fka5xHJo68gn-XVmQhxVZkxnPakGKwug,70144
+roksta/goal_workflow.cp314-win_amd64.pyd,sha256=t0MpBPUMGm2o1H7h95qiNyB-3u2nosr3XCw7Fj9fQx0,198144
+roksta/init_codebase.cp314-win_amd64.pyd,sha256=OUi3s0cSpMOx-mgDee4TEQhTC7lgJiV8pB4hlNSxD5w,117248
+roksta/lint_code.cp314-win_amd64.pyd,sha256=IXOeknKPgraTmMsv6uuKAt2RSIvhjkHLYi4yB5ALHNc,74240
+roksta/logger.cp314-win_amd64.pyd,sha256=PByehGy0Dq0nFJxnh9Vygo8D0BP0v10cJ5j9HhPPkQg,38400
+roksta/main.cp314-win_amd64.pyd,sha256=LeDTUK-V-3alzuG4w_d84S7Y9bQvqpePyDiDb64chzg,37888
+roksta/make_issue.cp314-win_amd64.pyd,sha256=YHhtGABOnWLrfw5eMm74YkfAzREIiX2MF7OMo-CEEsQ,68608
+roksta/new_features.cp314-win_amd64.pyd,sha256=8dFAnlgWmZ_x1uHE5chNpsJ5ysIgOtkkzhrpBdlT-GE,16896
+roksta/parse_readme.cp314-win_amd64.pyd,sha256=396ubyEMD5TXKhe-XTVxvpGpsVobbT0sYHXRUiY1dRQ,62976
+roksta/propose_solution.cp314-win_amd64.pyd,sha256=jOY5v5dZNV4drbLcOvBV_Ixe4lKTgZo7XKW4fGSdHc0,58880
+roksta/response_formats.cp314-win_amd64.pyd,sha256=_avk92XKPVe4_jVUmtbyrGsaAR6tTZL8iT_cd7_JsxY,38912
+roksta/rewrite_goal.cp314-win_amd64.pyd,sha256=aiYBGnCvWcYefuvnYRQ012nrw6_GW4omoj4A-_mU450,55808
+roksta/roksta.cp314-win_amd64.pyd,sha256=nMHWEmxJgB5gHz1EVTq1pJWE1gyZ5ixWqGWLyJ71EjM,233984
+roksta/run_cli_goal.cp314-win_amd64.pyd,sha256=h85oGVR4fDnRMQRW8MRgnkXKtPDd_5zmHNYvBkYioFs,68608
+roksta/select_files.cp314-win_amd64.pyd,sha256=kH7ZJabpaK_Rg-JKmvefFmJxTePHdgAeLjC1JdcegHw,63488
+roksta/tips.cp314-win_amd64.pyd,sha256=7Ume-47KRESQvv9uE9XhuNSSXI-0jNsEdG3OwZ6goe8,21504
+roksta/utils.cp314-win_amd64.pyd,sha256=0ZKbgGl11ac30p6zrbw4B3wTTA3OiM2MX4tyhqsiJoc,242688
+roksta/write_code.cp314-win_amd64.pyd,sha256=TsHc1FH1cGR6gOW2ZccWMGxZCaXEOSYqYuTje4G7EB4,56832
+roksta/ai/__init__.cp314-win_amd64.pyd,sha256=_U23mWgkukUaQgMOyREMvvDIQhpsF8EanNlOhpHXXco,16896
+roksta/ai/call_ai.cp314-win_amd64.pyd,sha256=vab6ACBEQasAZ3G8iwxnfQHTmjyKnQS6tLBMk9yGrow,74752
+roksta/ai/gemini.cp314-win_amd64.pyd,sha256=OzcrwuL0HeYFhQX6ZR832pTeui-irEtYi7zBJA-AtgA,148480
+roksta/ai/generic.cp314-win_amd64.pyd,sha256=IwYgOBDBAsDY--_rOjDG5wo8ic3VDIjlvIzJu-b_rjc,109568
+roksta/ai/llm.cp314-win_amd64.pyd,sha256=zDDeFGx9Rlst1vkocg83EijImuaJoEA-vFf2ih_xFXk,86528
+roksta/ai/openai.cp314-win_amd64.pyd,sha256=Gb3ZyVE6RKdPmWYeki5ezcq8teJsteeZdyC94ByCso4,105984
+roksta/ai/tools/__init__.cp314-win_amd64.pyd,sha256=yiiD3E0o_LNp5LYzwR6MD8KiFUFTe0YzH9BtoPWzVes,23552
+roksta/ai/tools/delete_file.cp314-win_amd64.pyd,sha256=R2s3YjoyRnDZiTPnlzmUEmOWCrttW_KjSLijvZPW8DA,34816
+roksta/ai/tools/edit_file.cp314-win_amd64.pyd,sha256=KofgRCfKssbF5D5D-KanD9M4r8SrT7myS4Awl66R8cA,46080
+roksta/ai/tools/final_response.cp314-win_amd64.pyd,sha256=LHxBBmkisJf_oDOSUp0Yhs_u2OMdbY7n-NMpKBIM_h0,22016
+roksta/ai/tools/get_file_summaries.cp314-win_amd64.pyd,sha256=ZApSf3gHpn2MoP5I4t2mHyNGDr7QrEBkGWWVjiG6rVc,36864
+roksta/ai/tools/read_file.cp314-win_amd64.pyd,sha256=cip7QWBaXDVIf7jQ6icTogURm-IbliLSfV3SgCK6EMU,46592
+roksta/ai/tools/regex_replace.cp314-win_amd64.pyd,sha256=1UDhPhFf8cM8NGwp3vjwXsru6FQ9RQhjL8ZPkT_GG6k,46080
+roksta/ai/tools/shell_any.cp314-win_amd64.pyd,sha256=--OajGsWAygbIN0FW2npWvkfyn5Goevl0TakD6hmFgg,70144
+roksta/ai/tools/shell_limited.cp314-win_amd64.pyd,sha256=Bqkn4o9qB1coBbdIQa-DChghbs2r2IVe3y_LrSKEVrg,57856
+roksta/ai/tools/tool_defs.cp314-win_amd64.pyd,sha256=yrsoXl8Wi99P53rPkwgwlUEArd6MR93NDcVuTjm7hSk,37888
+roksta/ai/tools/tool_utils.cp314-win_amd64.pyd,sha256=VUOF3pvA1S4FhE1oQf26K-NTjMuwzpwSsGFsn4aa3ao,100864
+roksta/ai/tools/write_file.cp314-win_amd64.pyd,sha256=0fSHZlBFVBCP_uCIczFXbYFNYri4Kzpmq0-Q0d1GdAQ,36352
+roksta/command_handlers/__init__.cp314-win_amd64.pyd,sha256=ITWkRkYrIeUpY3NtpNSxHF9gZj4gjUHETArNsfxAuuk,23040
+roksta/command_handlers/handle_activate_command.cp314-win_amd64.pyd,sha256=4RUJXTz2FBJo1p3oiGsh7xaog5xSi1q_ob_vp-SP56c,78848
+roksta/command_handlers/handle_add_funds_command.cp314-win_amd64.pyd,sha256=j1tFEuPC0fduxXMJgQcorxo-V2OpiEa-SzGg5q9q6m4,70144
+roksta/command_handlers/handle_auto_charge_command.cp314-win_amd64.pyd,sha256=ufxqVtc-AFc1meJOyJ2OZ0cQ6YXXO6AA0bmSH8epKow,51712
+roksta/command_handlers/handle_auto_commit_command.cp314-win_amd64.pyd,sha256=LOx29V3b1hGXbBUqWGOvpAiXn8wnfgSfJRLQQxTVYKw,37376
+roksta/command_handlers/handle_building_command.cp314-win_amd64.pyd,sha256=V3cvpHMO9ZXmVGqSoiz1lQHqf1upWfiZtnQpuhrWg1I,41984
+roksta/command_handlers/handle_chat_command.cp314-win_amd64.pyd,sha256=IYLc0Q3PpS60EMPiWvFvoEjZikJkMLFueRsirbdjdPs,41984
+roksta/command_handlers/handle_dev_rate_command.cp314-win_amd64.pyd,sha256=9iktxXoqUhlmXPzKxyHQmWfmDvBESqkJNq5z5GnDl8k,37376
+roksta/command_handlers/handle_feedback_command.cp314-win_amd64.pyd,sha256=bHWisuR0BhWawkn1bGMWSmbFQm19qcxWhRZbwqrm6ME,62976
+roksta/command_handlers/handle_goal_command.cp314-win_amd64.pyd,sha256=sqvtmFNE0ZCFIlI2pTwTkNfnyI175KjVWTXS66aXUQc,56320
+roksta/command_handlers/handle_help_command.cp314-win_amd64.pyd,sha256=YQP3OLDL9JQAlLo5cDLHncfCNqd0qz3EbZABZZqAsXw,75264
+roksta/command_handlers/handle_init_command.cp314-win_amd64.pyd,sha256=triFEX4fmkdcOl0bdFXU2PBMX_MghnnnGs-zYAi8dMc,53248
+roksta/command_handlers/handle_linting_command.cp314-win_amd64.pyd,sha256=ngPdYsSKsebzYWVpmNwTYZGJylW0npGheHX-RplVIgE,45056
+roksta/command_handlers/handle_login_command.cp314-win_amd64.pyd,sha256=NHnouhgYI1RW6wqYa4IhQ_5AcTb57NgzyFMyx5mvfz8,66048
+roksta/command_handlers/handle_logout_command.cp314-win_amd64.pyd,sha256=qt530leR7n4MX0MzDtioQCK_5cZWLKXCRRTKTSJHXbE,29184
+roksta/command_handlers/handle_payment_details_command.cp314-win_amd64.pyd,sha256=rBmx0CitiUeB23FWeYIFIhYbz3az8QouhzEOY_uPp-U,55808
+roksta/command_handlers/handle_quit_command.cp314-win_amd64.pyd,sha256=idGxZpN4AKgDCt-09RDBwJj5ZBRJdpG06HZNIJW9_XY,19968
+roksta/command_handlers/handle_redeem_command.cp314-win_amd64.pyd,sha256=mAzCCmt0qqLLwg55w1TzPFoCU_xswryxWuGb0gCG8nY,53248
+roksta/command_handlers/handle_request_activation_command.cp314-win_amd64.pyd,sha256=L2H6Pq5cOpQpS-M05bS9M3am9PbpIw1ljdigWvLKMC0,65536
+roksta/command_handlers/handle_testing_command.cp314-win_amd64.pyd,sha256=NYjXCnoMyYFt2ZcvW5YFq9fTHWyX2OsnF_duAM4lyak,41984
+roksta/command_handlers/handle_usage_command.cp314-win_amd64.pyd,sha256=fw_Xvp9SFsMrvBTeTf1qFaPwWR2nDA5mVlyc-3_VMGE,61952
+tests/__init__.py,sha256=meQjVtD2zimtDw3tIP6nNWHGDQVC42as-c_y038cYnk,126
+tests/conftest.py,sha256=4-DveP_c_DUwOcRu8_qAE8zJAWmc0kP6XeUXlWSCC8Q,8089
+tests/functions/__init__.py,sha256=msAiuVOA8jfqVxichPYYyfBFCVEFB_xRvj4lrSpKK1w,97
+tests/functions/test_auth.py,sha256=SNzmpkoxknrxx6cTE91MM2kN3F08GB1pRuupP5IrQs0,1016
+tests/functions/test_main.py,sha256=LgWLAOtSfDXNfHhAAr-j9Peb5JBLFtQ0eKAWqTxLNDU,2648
+tests/functions/test_utils.py,sha256=x-_30jai12k87c_7LwuYpk3_mGM_uTF6ZHXGvMWYlO8,17205
+tests/functions/api_v1_00/__init__.py,sha256=VdV5Oy6FRVT4S1xK34YodylJ7FY6jqvjujkS2Qvbyt0,126
+tests/functions/api_v1_00/test__analytics.py,sha256=s_pK4ySr86PC4ZBr7Y7bCHOEuFj36amaWrianBZ_QbU,16222
+tests/functions/api_v1_00/test__gemini_proxy.py,sha256=ZNYBQ9agPyNYGAVYlWPY6IF8wP_5IQ4nbHOpY8TnKEw,13696
+tests/functions/api_v1_00/test__generic_proxy.py,sha256=f2KjO8v_mh4nBwdkFXJLYfFUBpJS_RDtQNfn0ltE2oM,18101
+tests/functions/api_v1_00/test__get_payment_details.py,sha256=GRMAn8WR-07EBZ0IkeVE4MARFUbDdDhEHO2dITRxnVA,14091
+tests/functions/api_v1_00/test__openai_proxy.py,sha256=i1f7S8IFgWiG6WGG-YYptjTCOjSudujEMyhP16BVlW0,18175
+tests/functions/api_v1_00/test__redeem_credit_code.py,sha256=A5WwuSXBX5TiZPdYrDRkUmpppBXG00lcP0LmRz0gNr0,6514
+tests/functions/api_v1_00/test__sync_emails.py,sha256=e0rb637Ois25t__1G8Yp7Pj8WgC-R1I9SUgS6bK1dsU,13024
+tests/functions/api_v1_00/test__take_payment.py,sha256=RQBUHcgN3-XOoPtAjca_72q4BaZxr_L3G_Nss5-gb2g,18513
+tests/functions/api_v1_00/test__use_activation_code.py,sha256=ZOCwXHzenH0l1WhmZP8s8vaXIk58lHpKJ0igx0VEDwY,17398
+tests/functions/api_v1_01/__init__.py,sha256=VdV5Oy6FRVT4S1xK34YodylJ7FY6jqvjujkS2Qvbyt0,126
+tests/functions/api_v1_01/test__analytics.py,sha256=s_pK4ySr86PC4ZBr7Y7bCHOEuFj36amaWrianBZ_QbU,16222
+tests/functions/api_v1_01/test__gemini_proxy.py,sha256=ZNYBQ9agPyNYGAVYlWPY6IF8wP_5IQ4nbHOpY8TnKEw,13696
+tests/functions/api_v1_01/test__generic_proxy.py,sha256=f2KjO8v_mh4nBwdkFXJLYfFUBpJS_RDtQNfn0ltE2oM,18101
+tests/functions/api_v1_01/test__get_payment_details.py,sha256=GRMAn8WR-07EBZ0IkeVE4MARFUbDdDhEHO2dITRxnVA,14091
+tests/functions/api_v1_01/test__openai_proxy.py,sha256=i1f7S8IFgWiG6WGG-YYptjTCOjSudujEMyhP16BVlW0,18175
+tests/functions/api_v1_01/test__redeem_credit_code.py,sha256=A5WwuSXBX5TiZPdYrDRkUmpppBXG00lcP0LmRz0gNr0,6514
+tests/functions/api_v1_01/test__sync_emails.py,sha256=e0rb637Ois25t__1G8Yp7Pj8WgC-R1I9SUgS6bK1dsU,13024
+tests/functions/api_v1_01/test__take_payment.py,sha256=RQBUHcgN3-XOoPtAjca_72q4BaZxr_L3G_Nss5-gb2g,18513
+tests/functions/api_v1_01/test__use_activation_code.py,sha256=ZOCwXHzenH0l1WhmZP8s8vaXIk58lHpKJ0igx0VEDwY,17398
+tests/functions/api_v1_02/__init__.py,sha256=VdV5Oy6FRVT4S1xK34YodylJ7FY6jqvjujkS2Qvbyt0,126
+tests/functions/api_v1_02/test__analytics.py,sha256=s_pK4ySr86PC4ZBr7Y7bCHOEuFj36amaWrianBZ_QbU,16222
+tests/functions/api_v1_02/test__gemini_proxy.py,sha256=ZNYBQ9agPyNYGAVYlWPY6IF8wP_5IQ4nbHOpY8TnKEw,13696
+tests/functions/api_v1_02/test__generic_proxy.py,sha256=f2KjO8v_mh4nBwdkFXJLYfFUBpJS_RDtQNfn0ltE2oM,18101
+tests/functions/api_v1_02/test__get_payment_details.py,sha256=GRMAn8WR-07EBZ0IkeVE4MARFUbDdDhEHO2dITRxnVA,14091
+tests/functions/api_v1_02/test__openai_proxy.py,sha256=i1f7S8IFgWiG6WGG-YYptjTCOjSudujEMyhP16BVlW0,18175
+tests/functions/api_v1_02/test__redeem_credit_code.py,sha256=A5WwuSXBX5TiZPdYrDRkUmpppBXG00lcP0LmRz0gNr0,6514
+tests/functions/api_v1_02/test__sync_emails.py,sha256=e0rb637Ois25t__1G8Yp7Pj8WgC-R1I9SUgS6bK1dsU,13024
+tests/functions/api_v1_02/test__take_payment.py,sha256=RQBUHcgN3-XOoPtAjca_72q4BaZxr_L3G_Nss5-gb2g,18513
+tests/functions/api_v1_02/test__use_activation_code.py,sha256=ZOCwXHzenH0l1WhmZP8s8vaXIk58lHpKJ0igx0VEDwY,17398
+tests/functions/api_v1_02/test_proxy_keyword_replacement.py,sha256=NFr_2f5mYinqx1pwdQjSL4MUs6Zge6Kzz4SeX8H8xq4,24189
+tests/functions/api_v1_02/test_replace_keywords.py,sha256=pFl3XWxujSwkXcuSxsznv_8qvx4u2rh441V6U-31yvc,2757
+roksta-0.3.2.dist-info/METADATA,sha256=NBLudPACgAEsf0RubYaDnGy7TqzoS6WcXzNtq5Dra3o,1496
+roksta-0.3.2.dist-info/WHEEL,sha256=7k6Wcy588iJYe5lf5K095NLg-uoBTnE-T8eHJ92G4_4,101
+roksta-0.3.2.dist-info/entry_points.txt,sha256=mzRdYg_DlzZRwjxYUt9-gyoRCkM1QBTeTbwETgiTdGw,44
+roksta-0.3.2.dist-info/top_level.txt,sha256=lvciNZQ1dPGXpiCLdWVXK03n9fKHjbQdwjqQbnUjeYM,13
+roksta-0.3.2.dist-info/RECORD,,

tests/conftest.py

@@ -167,3 +167,45 @@ except Exception:
     _attach_gtypes_stubs(gtypes)
     genai.types = gtypes
     sys.modules['google.genai.types'] = gtypes
+
+# Provide a minimal nacl.public stub if PyNaCl is not installed. This prevents
+# import-time failures in tests that import `roksta.firebase` which does
+# `from nacl.public import PublicKey, SealedBox`.
+try:
+    import nacl.public  # type: ignore
+except Exception:
+    # Create lightweight stub modules to satisfy `from nacl.public import PublicKey, SealedBox`
+    nacl = _types_mod.ModuleType('nacl')
+    nacl_public = _types_mod.ModuleType('nacl.public')
+
+    class PublicKey:
+        def __init__(self, data):
+            # Accept bytes-like input; ensure it's bytes when possible.
+            try:
+                self._data = bytes(data)
+            except Exception:
+                self._data = data
+
+        def __repr__(self):
+            try:
+                length = len(self._data)
+            except Exception:
+                length = 'unknown'
+            return f"<PublicKey len={length}>"
+
+    class SealedBox:
+        def __init__(self, public_key):
+            self.public_key = public_key
+
+        def encrypt(self, plaintext: bytes) -> bytes:
+            # Return deterministic bytes so callers can base64-encode the result.
+            if not isinstance(plaintext, (bytes, bytearray)):
+                plaintext = str(plaintext).encode('utf-8')
+            return b"STUB_ENCRYPTED:" + bytes(plaintext)
+
+    nacl_public.PublicKey = PublicKey
+    nacl_public.SealedBox = SealedBox
+    # Attach the submodule on the parent module object for attribute access
+    nacl.public = nacl_public
+    sys.modules['nacl'] = nacl
+    sys.modules['nacl.public'] = nacl_public

tests/functions/{api_v0_01 → api_v1_01}/__init__.py

@@ -1,2 +1,2 @@
-# tests.functions.api_v0_01 package marker
+# tests.functions.api_v1_00 package marker
 # Allows tests under this directory to be imported with package-qualified names.

tests/functions/{api_v0_01 → api_v1_01}/test__analytics.py

@@ -176,12 +176,11 @@ sys.modules['httpx'] = httpx_mod
 # -----------------------------
 repo_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..', '..'))
 functions_root = os.path.join(repo_root, 'functions')
-module_path = os.path.join(functions_root, 'api_v0_01', '_analytics.py')
-spec = importlib.util.spec_from_file_location('api_v0_01._analytics', module_path)
+module_path = os.path.join(functions_root, 'api_v1_00', '_analytics.py')
+spec = importlib.util.spec_from_file_location('api_v1_00._analytics', module_path)
 _analytics = importlib.util.module_from_spec(spec)
 spec.loader.exec_module(_analytics)
 
-
 # Restore original sys.modules mappings to avoid side-effects for other tests
 for name, orig in _orig_sys_modules.items():
     if orig is None:

tests/functions/{api_v0_01 → api_v1_01}/test__gemini_proxy.py

@@ -19,6 +19,8 @@ if FUNCTIONS_DIR not in sys.path:
 _orig_sys_modules = {}
 _names_to_fake = [
     'firebase_functions',
+    'firebase_admin',
+    'firebase_admin.firestore',
     'utils',
     'auth',
     'google',
@@ -49,6 +51,35 @@ class FakeResponse:
 firebase_functions.https_fn = types.SimpleNamespace(Request=object, Response=FakeResponse)
 sys.modules['firebase_functions'] = firebase_functions
 
+# Fake firebase_admin and its firestore submodule to satisfy optional imports in the proxy
+firebase_admin = types.ModuleType('firebase_admin')
+firestore_mod = types.ModuleType('firebase_admin.firestore')
+
+# Minimal Client type for annotations and a no-op client() factory
+class _DummyClient:  # pragma: no cover - just a stub for import-time type hints
+    pass
+
+def _dummy_client():
+    # Return a simple object; logic using it is patched in tests
+    return types.SimpleNamespace()
+
+# Decorator stub used by perform_atomic_debit; never actually invoked in tests
+def _transactional(fn=None):
+    def wrapper(*args, **kwargs):
+        return fn(*args, **kwargs)
+    return wrapper
+
+# Populate stubs on the firestore module
+setattr(firestore_mod, 'Client', _DummyClient)
+setattr(firestore_mod, 'client', _dummy_client)
+setattr(firestore_mod, 'transactional', _transactional)
+setattr(firestore_mod, 'SERVER_TIMESTAMP', object())
+
+# Attach firestore submodule to firebase_admin package and register in sys.modules
+setattr(firebase_admin, 'firestore', firestore_mod)
+sys.modules['firebase_admin'] = firebase_admin
+sys.modules['firebase_admin.firestore'] = firestore_mod
+
 # Fake utils module (provides functions imported by _gemini_proxy)
 utils_mod = types.ModuleType('utils')
 
@@ -118,14 +149,14 @@ sys.modules['google.genai'] = genai_mod
 sys.modules['google.genai.types'] = types_mod
 
 # Import the module under test after preparing the fake imports
+# Resolve the actual repository 'functions' directory (tests files live under tests/, so climb up to repo root)
 repo_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..', '..'))
 functions_root = os.path.join(repo_root, 'functions')
-module_path = os.path.join(functions_root, 'api_v0_01', '_gemini_proxy.py')
-spec = importlib.util.spec_from_file_location('api_v0_01._gemini_proxy', module_path)
+module_path = os.path.join(functions_root, 'api_v1_00', '_gemini_proxy.py')
+spec = importlib.util.spec_from_file_location('api_v1_00._gemini_proxy', module_path)
 _gemini = importlib.util.module_from_spec(spec)
 spec.loader.exec_module(_gemini)
 
-
 # Restore original sys.modules mappings to avoid side-effects for other tests
 for name, orig in _orig_sys_modules.items():
     if orig is None:
@@ -198,7 +229,9 @@ def test_invalid_auth_key_returns_403():
 def test_non_post_method_returns_405():
     headers = {_gemini.AUTH_HEADER_NAME: 'ok'}
     req = DummyRequest(headers=headers, method='GET')
-    with patch.object(_gemini, 'validate_auth_key', return_value=True), patch.object(_gemini, 'verify_firebase_token', return_value={}):
+    with patch.object(_gemini, 'validate_auth_key', return_value=True), \
+         patch.object(_gemini, 'verify_firebase_token', return_value={}), \
+         patch.object(_gemini, 'ensure_balance_positive', return_value=(True, 100.0)):
         resp = _gemini._gemini_proxy(req)
 
     assert resp.status_code == 405
@@ -210,7 +243,9 @@ def test_non_post_method_returns_405():
 def test_malformed_json_returns_400():
     headers = {_gemini.AUTH_HEADER_NAME: 'ok'}
     req = DummyRequest(headers=headers, method='POST', raise_on_get_json=True)
-    with patch.object(_gemini, 'validate_auth_key', return_value=True), patch.object(_gemini, 'verify_firebase_token', return_value={}):
+    with patch.object(_gemini, 'validate_auth_key', return_value=True), \
+         patch.object(_gemini, 'verify_firebase_token', return_value={}), \
+         patch.object(_gemini, 'ensure_balance_positive', return_value=(True, 100.0)):
         resp = _gemini._gemini_proxy(req)
 
     assert resp.status_code == 400
@@ -223,7 +258,9 @@ def test_rehydrate_params_error_returns_400():
     headers = {_gemini.AUTH_HEADER_NAME: 'ok'}
     # missing 'model' to cause KeyError inside rehydrate_params
     req = DummyRequest(headers=headers, method='POST', json_data={'contents': 'x', 'config': {}})
-    with patch.object(_gemini, 'validate_auth_key', return_value=True), patch.object(_gemini, 'verify_firebase_token', return_value={}):
+    with patch.object(_gemini, 'validate_auth_key', return_value=True), \
+         patch.object(_gemini, 'verify_firebase_token', return_value={}), \
+         patch.object(_gemini, 'ensure_balance_positive', return_value=(True, 100.0)):
         resp = _gemini._gemini_proxy(req)
 
     assert resp.status_code == 400
@@ -237,6 +274,7 @@ def test_get_api_key_failure_returns_500():
     req = DummyRequest(headers=headers, method='POST', json_data={'model': 'g', 'contents': 'hi', 'config': {}})
     with patch.object(_gemini, 'validate_auth_key', return_value=True), \
          patch.object(_gemini, 'verify_firebase_token', return_value={}), \
+         patch.object(_gemini, 'ensure_balance_positive', return_value=(True, 100.0)), \
          patch.object(_gemini, 'get_api_key', side_effect=Exception('boom')):
         resp = _gemini._gemini_proxy(req)
 
@@ -258,12 +296,18 @@ def test_successful_flow_calls_genai_and_returns_payload():
 
         def generate_content(self, **params):
             class FakeResp:
+                def __init__(self):
+                    # include usage metadata expected by get_usage()
+                    self.usage_metadata = types.SimpleNamespace(prompt_token_count=1, total_token_count=2)
+
                 def to_json_dict(self_inner):
                     return { 'result': 'ok', 'received_model': params.get('model') }
             return FakeResp()
 
     with patch.object(_gemini, 'validate_auth_key', return_value=True), \
          patch.object(_gemini, 'verify_firebase_token', return_value={}), \
+         patch.object(_gemini, 'ensure_balance_positive', return_value=(True, 100.0)), \
+         patch.object(_gemini, 'bill_with_retry', return_value=("ok", 99.5)), \
          patch.object(_gemini, 'get_api_key', return_value='GEMINI-KEY'), \
          patch.object(_gemini.genai, 'Client', FakeClient):
         resp = _gemini._gemini_proxy(req)
@@ -297,6 +341,7 @@ def test_genai_exception_with_status_code_is_returned_as_error_status():
 
     with patch.object(_gemini, 'validate_auth_key', return_value=True), \
          patch.object(_gemini, 'verify_firebase_token', return_value={}), \
+         patch.object(_gemini, 'ensure_balance_positive', return_value=(True, 100.0)), \
          patch.object(_gemini, 'get_api_key', return_value='GEMINI-KEY'), \
          patch.object(_gemini.genai, 'Client', ErrClient):
         resp = _gemini._gemini_proxy(req)

tests/functions/{api_v0_01 → api_v1_01}/test__generic_proxy.py

@@ -108,13 +108,42 @@ openai_mod.APIError = APIError
 sys.modules['openai'] = openai_mod
 
 # Import the module under test after preparing the fake imports
+#_generic = importlib.import_module('_generic_proxy')
+
 repo_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..', '..'))
 functions_root = os.path.join(repo_root, 'functions')
-module_path = os.path.join(functions_root, 'api_v0_01', '_generic_proxy.py')
-spec = importlib.util.spec_from_file_location('api_v0_01._generic_proxy', module_path)
+module_path = os.path.join(functions_root, 'api_v1_00', '_generic_proxy.py')
+spec = importlib.util.spec_from_file_location('api_v1_00._generic_proxy', module_path)
 _generic = importlib.util.module_from_spec(spec)
 spec.loader.exec_module(_generic)
 
+# Patch Firestore client and billing functions to avoid external dependencies during tests
+# Provide a dummy Firestore client and make balance checks/billing no-ops.
+_generic.firestore = types.SimpleNamespace(client=lambda: object())
+
+def _fake_ensure_balance_positive(db, user_id):
+    return True, 100.0
+
+def _fake_extract_usage(llm_family=None, call_type=None, model_id=None, request_payload=None, response_payload=None):
+    # Default to zero-usage unless present in the payload
+    usage = response_payload.get('usage', {}) if isinstance(response_payload, dict) else {}
+    in_tokens = usage.get('prompt_tokens', 0) or usage.get('input_tokens', 0) or 0
+    out_tokens = usage.get('completion_tokens', 0) or usage.get('output_tokens', 0) or 0
+    return {"input_tokens": in_tokens, "output_tokens": out_tokens}
+
+def _fake_calculate_cost(model_id=None, input_tokens=0, output_tokens=0):
+    return 0.0
+
+def _fake_bill_with_retry(db=None, user_id=None, model_id=None, usage=None, cost=None, reason=None):
+    return "ok", 100.0
+
+_generic.ensure_balance_positive = _fake_ensure_balance_positive
+_generic.extract_usage = _fake_extract_usage
+# Compatibility shim: some versions of the module call get_usage(response) instead of extract_usage(...)
+_generic.get_usage = lambda response: _fake_extract_usage(response_payload=response)
+_generic.calculate_cost = _fake_calculate_cost
+_generic.bill_with_retry = _fake_bill_with_retry
+
 # Restore original sys.modules mappings to avoid side-effects for other tests
 for name, orig in _orig_sys_modules.items():
     if orig is None:

tests/functions/{api_v0_01 → api_v1_01}/test__get_payment_details.py

@@ -186,8 +186,8 @@ sys.modules['ulid'] = ulid_mod
 # Load the module directly from the functions/ tree so the test's fake sys.modules entries are respected
 repo_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..', '..'))
 functions_root = os.path.join(repo_root, 'functions')
-module_path = os.path.join(functions_root, 'api_v0_01', '_get_payment_details.py')
-spec = importlib.util.spec_from_file_location('api_v0_01._get_payment_details', module_path)
+module_path = os.path.join(functions_root, 'api_v1_00', '_get_payment_details.py')
+spec = importlib.util.spec_from_file_location('api_v1_00._get_payment_details', module_path)
 _get_payment_details = importlib.util.module_from_spec(spec)
 spec.loader.exec_module(_get_payment_details)