Skip to content

Commit 17aabc0

Browse files
committed
feat: first working version of payments for nilAI
1 parent cd6581c commit 17aabc0

File tree

3 files changed

+18
-0
lines changed

3 files changed

+18
-0
lines changed

docker-compose.dev.yml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -106,11 +106,16 @@ services:
106106
depends_on:
107107
nilauth-postgres:
108108
condition: service_healthy
109+
<<<<<<< HEAD
109110
healthcheck:
110111
test: ["CMD", "wget", "--no-verbose", "--tries=1", "--spider", "http://localhost:3000/health"]
111112
interval: 30s
112113
retries: 3
113114
start_period: 15s
114115
timeout: 10s
116+
=======
117+
118+
119+
>>>>>>> a4d2f92 (feat: first working version of payments for nilAI)
115120
volumes:
116121
postgres_data:

nilai-api/pyproject.toml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,10 @@ dependencies = [
3636
"pyyaml>=6.0.1",
3737
"trafilatura>=1.7.0",
3838
"secretvaults",
39+
<<<<<<< HEAD
3940
"e2b-code-interpreter>=1.0.3",
41+
=======
42+
>>>>>>> a4d2f92 (feat: first working version of payments for nilAI)
4043
"nilauth-credit-middleware>=0.1.0",
4144
]
4245

nilai-api/src/nilai_api/credit.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,11 @@
1212

1313
from nilai_api.config import CONFIG
1414

15+
<<<<<<< HEAD
1516
from nuc.envelope import NucTokenEnvelope
1617

18+
=======
19+
>>>>>>> a4d2f92 (feat: first working version of payments for nilAI)
1720
logger = logging.getLogger(__name__)
1821

1922

@@ -94,7 +97,11 @@ class LLMResponse(BaseModel):
9497

9598
def user_id_extractor() -> Callable[[Request], Awaitable[str]]:
9699
if CONFIG.auth.auth_strategy == "nuc":
100+
<<<<<<< HEAD
97101
return from_nuc_bearer_root_token()
102+
=======
103+
return UserIdExtractors.from_nuc_bearer_token()
104+
>>>>>>> a4d2f92 (feat: first working version of payments for nilAI)
98105
else:
99106
extractor = UserIdExtractors.from_header("Authorization")
100107

@@ -109,6 +116,7 @@ async def wrapper(request: Request) -> str:
109116
return wrapper
110117

111118

119+
<<<<<<< HEAD
112120
def from_nuc_bearer_root_token() -> Callable[[Request], Awaitable[str]]:
113121
"""Extract user ID from a NUC root token"""
114122

@@ -127,6 +135,8 @@ async def extractor(request: Request) -> str:
127135
return extractor
128136

129137

138+
=======
139+
>>>>>>> a4d2f92 (feat: first working version of payments for nilAI)
130140
def llm_cost_calculator(llm_cost_dict: LLMCostDict):
131141
async def calculator(request: Request, response_data: dict) -> float:
132142
model_name = getattr(request, "model", "default")

0 commit comments

Comments
 (0)