Create a helper function to poll for knowledge base database creation

[bbatha]

When a Gradient AI Platform Knowledge Base is created the deployment of the database can run for several minutes. Currently users must write loops to poll for the deployment state, roughly.

from gradient import Gradient

client = Gradient()
while True:
  knowledge_base = client.knowledge_bases.retrieve(
      "uuid",
  )
  if knowledge_base.database_status != "ONLINE"
    continue

This should of course handle failed states for errors and such

[sumeet156]

Hi @bbatha 👋, I’d like to work on this issue for HacktoberFest.

My plan is to create a helper function that automatically waits for the Knowledge Base to reach the ONLINE state, so users don’t have to write their own polling loops.

I’ll make sure to:

• Handle failed or error states properly
• Add a timeout or retry mechanism to avoid infinite loops
• Keep the code simple and well-documented

Please assign this issue to me. I’m excited to learn and contribute. Thank you! 🙌

[shreya112003]

Hi @bbatha 👋
I’d love to work on this issue as part of Hacktoberfest! I’m a final-year B.Tech student exploring open-source contributions for the first time and this looks like a great opportunity to start.

Approach:

Implement a helper function (e.g., poll_for_knowledge_base_ready) that continuously checks the deployment status of a Knowledge Base.

Use an interval-based polling mechanism with exponential backoff or a configurable sleep time.

Gracefully handle terminal states such as "FAILED" or "ERROR" to avoid infinite loops.

Return once the status reaches "ONLINE" or raise an informative exception otherwise.

Add minimal tests and docstrings to ensure clarity and maintainability.

If this sounds good, could you please assign the issue to me? I’ll make sure to follow contribution guidelines and open a PR soon.

Thanks a lot! 😊

[dhruv-git-sys]

Can u assign this issue to me?

[MimoHasPurpose]

@bbatha I wanna work on this. could u assign me?

[JsMelix]

assign me please for hacktober :)

[DaveKavya]

Hi! I’d love to work on this issue. Could you please assign it to me?