Gen-AI python implementation

[Jeel-mehta]

Issue #, if available:

Description of changes:

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

[yiyuan-he]

Workflow builds are failing. Could you take a look?

They are probably just linting failures which you can resolve by following the instructions under Code Style Check here

[yiyuan-he]

Trying to access element with [0] will throw an error if the results array doesn't exist. I think we need another conditional check here.

For example:

if 'results' in response_body:
   result = response_body["results"][0]

or something more concise like:

result = response_body.get('results', [{}])

[Jeel-mehta]

I just saw this, I will fix this and make another commit

[yiyuan-he]

I think this logic will still throw an error.

For example, if results does not exist in the response_body, the checks will still continue to the other conditionals like if "tokenCount" in result:, which will then give an error since result will be unbound.

You can do something like this instead:

if "results" in response_body and response_body["results"]:
      result = response_body["results"][0]
      if "tokenCount" in result:
          span.set_attribute(GEN_AI_USAGE_OUTPUT_TOKENS, result["tokenCount"])
      if "completionReason" in result:
          ...

[Jeel-mehta]

I fixed this

[yiyuan-he]

Looks good. Thanks!