Skip to content

Commit

Permalink
clean up
Browse files Browse the repository at this point in the history
  • Loading branch information
Josh-XT committed Jan 7, 2025
1 parent 8c57b95 commit 5a8bc33
Showing 1 changed file with 13 additions and 17 deletions.
30 changes: 13 additions & 17 deletions agixt/graphqlendpoints/Conversations.py
Original file line number Diff line number Diff line change
Expand Up @@ -260,7 +260,7 @@ class NotificationEvent:
@strawberry.type
class ProviderSetting:
name: str
value: Optional[str]
default_value: Optional[str]


@strawberry.type
Expand Down Expand Up @@ -317,7 +317,7 @@ class EmbedderList:
def convert_settings_to_type(settings_dict: Dict[str, str]) -> List[ProviderSetting]:
"""Convert settings dictionary to list of ProviderSetting objects"""
return [
ProviderSetting(name=key, value=str(value))
ProviderSetting(name=key, default_value=str(value))
for key, value in settings_dict.items()
]

Expand Down Expand Up @@ -532,28 +532,24 @@ async def notifications(
page_info=page_info, edges=notifications[start_idx:end_idx]
)

@strawberry.field
async def providers(self, info) -> List[ProviderWithSettings]:
"""Get all providers with their settings"""
user = await get_user_from_context(info)
providers_settings = get_providers_with_settings()
return [
ProviderWithSettings(
provider=Provider(
name=list(provider.keys())[0],
settings=convert_settings_to_type(list(provider.values())[0]),
)
)
for provider in providers_settings
]

@strawberry.field
async def providers_by_service(self, info, service: str) -> ProviderList:
"""Get providers that offer a specific service"""
user = await get_user_from_context(info)
providers = get_providers_by_service(service=service)
return ProviderList(providers=providers)

@strawberry.field
async def providers(self, info) -> ProvidersWithDetails:
"""Get comprehensive provider details"""
user = await get_user_from_context(info)
provider_details = get_providers_with_details()
providers = [
convert_provider_details({"name": name, **details})
for name, details in provider_details.items()
]
return ProvidersWithDetails(providers=providers)


# Response types for mutations
@strawberry.type
Expand Down

0 comments on commit 5a8bc33

Please sign in to comment.